HomeSort by relevance Sort by last modified time
    Searched refs:input_batch_size (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
batchtospace_op.cc 139 int64 input_batch_size = orig_input_batch_size; local
142 input_batch_size *= size;
145 internal_input_shape.AddDim(input_batch_size);
146 internal_output_shape.AddDim(input_batch_size / block_shape_product);
spacetobatch_op.cc 136 int64 input_batch_size = orig_input_tensor.dim_size(0); local
139 input_batch_size *= size;
142 internal_input_shape.AddDim(input_batch_size);
143 internal_output_shape.AddDim(input_batch_size * block_shape_product);
  /external/tensorflow/tensorflow/contrib/training/python/training/
sampling_ops_test.py 186 input_batch_size = 11
187 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)
  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
reference_ops.h     [all...]
  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
optimized_ops.h     [all...]

Completed in 318 milliseconds