Searched defs:input_batch_size (Results 1 – 7 of 7) sorted by relevance
139 int64 input_batch_size = orig_input_batch_size; in BatchToSpaceOpCompute() local
137 int64 input_batch_size = orig_input_tensor.dim_size(0); in SpaceToBatchOpCompute() local
1083 const int64 input_batch_size = in HandleConvolution() local
1894 const int input_batch_size = 1) { in TestBinaryTensorOpWeightFallback()
3092 const int input_batch_size = input1_shape.Dims(0); in SpaceToBatchND() local3154 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local
5800 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local
6128 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local