Home
last modified time | relevance | path

Searched defs:input_batch_dim (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc41 auto input_batch_dim = dnums.input_batch_dimension(); in Run() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dbatch_seq_config_test.cc108 TfLiteIntArray* input_batch_dim = in ApplyDelegate() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter.cc181 auto input_batch_dim = conv_dnums.input_batch_dimension(); in MatchBackwardFilter() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc579 const int64 input_batch_dim = dnums.input_batch_dimension(); in HandleConvolution() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2590 const int input_batch_dim = input_tensor.batch_size(); in ConvertReshape() local