Searched refs:input_batch_size (Results 1 – 9 of 9) sorted by relevance
139 int64 input_batch_size = orig_input_batch_size; in BatchToSpaceOpCompute() local142 input_batch_size *= size; in BatchToSpaceOpCompute()145 internal_input_shape.AddDim(input_batch_size); in BatchToSpaceOpCompute()146 internal_output_shape.AddDim(input_batch_size / block_shape_product); in BatchToSpaceOpCompute()
137 int64 input_batch_size = orig_input_tensor.dim_size(0); in SpaceToBatchOpCompute() local140 input_batch_size *= size; in SpaceToBatchOpCompute()143 internal_input_shape.AddDim(input_batch_size); in SpaceToBatchOpCompute()144 internal_output_shape.AddDim(input_batch_size * block_shape_product); in SpaceToBatchOpCompute()
186 input_batch_size = 11187 val_input_batch = [array_ops.zeros([input_batch_size, 2, 3, 4])]190 lambda: array_ops.ones([input_batch_size], dtype=dtypes.int32) * 1,191 lambda: array_ops.ones([input_batch_size], dtype=dtypes.int32) * 3)
3092 const int input_batch_size = input1_shape.Dims(0); in SpaceToBatchND() local3107 int input_batch = out_b % input_batch_size; in SpaceToBatchND()3108 int shift_w = (out_b / input_batch_size) % block_shape_width; in SpaceToBatchND()3109 int shift_h = (out_b / input_batch_size) / block_shape_width; in SpaceToBatchND()3154 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local3161 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND()
1083 const int64 input_batch_size = in HandleConvolution() local1086 const int64 batch_group_size = input_batch_size / batch_group_count; in HandleConvolution()1175 ((batch_group_index * batch_group_size) % input_batch_size) * in HandleConvolution()
1894 const int input_batch_size = 1) { in TestBinaryTensorOpWeightFallback() argument1903 test->AddTestTensor("input", /*dims=*/input_dims, input_batch_size, in TestBinaryTensorOpWeightFallback()
3592 const int64_t input_batch_size = params.input_row_size * params.input_height;3624 const uint8* input_ptr = input_data + b * input_batch_size;
5800 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local5807 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND()
6128 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local6135 for (int in_batch = 0; in_batch < input_batch_size; ++in_batch) { in BatchToSpaceND()