Searched defs:input_batch_dim (Results 1 – 7 of 7) sorted by relevance
43 auto input_batch_dim = dnums.input_batch_dimension(); in Run() local
46 int64_t input_batch_dim; member
102 auto input_batch_dim = conv_dnums.input_batch_dimension(); in MatchBackwardFilter() local
109 TfLiteIntArray* input_batch_dim = in ApplyDelegate() local
597 const int64_t input_batch_dim = dnums.input_batch_dimension(); in GetConvolutionFlops() local
3478 const int64_t input_batch_dim = in PartitionDot() local
2218 const int input_batch_dim = input_tensor.batch_size(); in ConvertStaticReshapeForImplicitBatchMode() local