Home
last modified time | relevance | path

Searched defs:output_batch_dim (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dconv_canonicalization.cc105 auto output_batch_dim = dnums.output_batch_dimension(); in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Djitrt_custom_calls.h54 int64_t output_batch_dim; member
H A Dgpu_conv_rewriter.cc108 auto output_batch_dim = conv_dnums.output_batch_dimension(); in MatchBackwardFilter() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dbatch_seq_config_test.cc111 TfLiteIntArray* output_batch_dim = in ApplyDelegate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dspace_to_batch_converter.cc3811 const int64_t output_batch_dim = new_dim_numbers.output_batch_dimension(); in PerformSpaceToBatchOnConvolution() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc2219 const int64_t output_batch_dim = output_dims.dim(0); in ConvertStaticReshapeForImplicitBatchMode() local