Home
last modified time | relevance | path

Searched defs:expanded_input (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dloss.h120 auto expanded_input = broadcast_tensors[0]; in mse_loss() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dprepare_tf.cc1280 TF::ExpandDimsOp expanded_input = rewriter.create<TF::ExpandDimsOp>( in matchAndRewrite() local