/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | runtime_fft_impl.h | 36 int64 input_batch, int64 fft_length0, int64 fft_length1, in EigenFftC2C() argument 46 dims[0] = input_batch; in EigenFftC2C() 63 int64 input_batch, int64 fft_length0, int64 fft_length1, in EigenFftR2C() argument 69 in_dims[0] = input_batch; in EigenFftR2C() 71 out_dims[0] = input_batch; in EigenFftR2C() 102 int64 input_batch, int64 fft_length0, int64 fft_length1, in EigenFftC2R() argument 108 in_dims[0] = input_batch; in EigenFftC2R() 110 out_dims[0] = input_batch; in EigenFftC2R() 173 int32 fft_type, int64 input_batch, int64 fft_length0, in EigenFftWithRank() argument 179 static_cast<complex64*>(operand), input_batch, fft_length0, in EigenFftWithRank() [all …]
|
D | runtime_conv2d_impl.h | 29 ScalarType* rhs, int64 input_batch, int64 input_rows, in EigenConvImpl() argument 40 input(lhs, input_batch, input_rows, input_cols, input_channels); in EigenConvImpl() 48 output(out, input_batch, output_rows, output_cols, kernel_filters); in EigenConvImpl() 58 pre_contract_dims[0] = output_cols * output_rows * input_batch; in EigenConvImpl() 63 post_contract_dims[0] = input_batch; in EigenConvImpl()
|
D | runtime_single_threaded_conv2d.cc | 27 Eigen::half* rhs, int64 input_batch, int64 input_rows, int64 input_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument 35 Eigen::DefaultDevice(), out, lhs, rhs, input_batch, input_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF16() 45 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument 52 Eigen::DefaultDevice(), out, lhs, rhs, input_batch, input_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
D | runtime_conv2d_mkl.cc | 53 ScalarType* rhs, int64 input_batch, int64 input_rows, in MKLConvImpl() argument 68 memory::dims conv1_src_dim = {ToInt(input_batch), ToInt(input_channels), in MKLConvImpl() 73 memory::dims conv1_dst_dim = {ToInt(input_batch), ToInt(kernel_filters), in MKLConvImpl() 154 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_MKLConvF32() argument 165 run_options_ptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_MKLConvF32() 171 MKLConvImpl(nullptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_MKLConvF32()
|
D | runtime_conv2d.cc | 29 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_EigenConvF32() argument 38 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConvF32() 47 Eigen::half* rhs, int64 input_batch, int64 input_rows, int64 input_cols, in __xla_cpu_runtime_EigenConvF16() argument 57 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConvF16()
|
D | runtime_single_threaded_fft.cc | 27 int32 fft_rank, int64 input_batch, int64 fft_length0, int64 fft_length1, in __xla_cpu_runtime_EigenSingleThreadedFft() argument 30 fft_rank, input_batch, fft_length0, fft_length1, in __xla_cpu_runtime_EigenSingleThreadedFft()
|
D | runtime_fft.cc | 30 int32 fft_rank, int64 input_batch, int64 fft_length0, int64 fft_length1, in __xla_cpu_runtime_EigenFft() argument 35 operand, fft_type, fft_rank, input_batch, in __xla_cpu_runtime_EigenFft()
|
D | runtime_single_threaded_conv2d.h | 27 tensorflow::int64 input_batch, tensorflow::int64 input_rows, 40 float* lhs, float* rhs, tensorflow::int64 input_batch,
|
D | runtime_conv2d.h | 26 float* lhs, float* rhs, tensorflow::int64 input_batch, 41 tensorflow::int64 input_batch, tensorflow::int64 input_rows,
|
D | runtime_fft.h | 26 tensorflow::int64 input_batch, tensorflow::int64 fft_length0,
|
D | runtime_single_threaded_fft.h | 26 tensorflow::int64 input_batch, tensorflow::int64 fft_length0,
|
D | runtime_conv2d_mkl.h | 26 float* lhs, float* rhs, tensorflow::int64 input_batch,
|
D | ir_emitter.cc | 1125 int64 input_batch = input_shape.dimensions(dnums.input_batch_dimension()); in HandleConvolution() local 1221 b_.getInt64(input_batch), in HandleConvolution() 1270 int64 input_batch = 1; in HandleFft() local 1272 input_batch *= fft->shape().dimensions(i); in HandleFft() 1301 b_.getInt32(fft_rank), b_.getInt64(input_batch), in HandleFft()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | quantized_batch_norm_op_test.cc | 63 const int input_batch = 1; in TEST_F() local 68 {input_batch, input_height, input_width, input_depth}); in TEST_F() 122 TensorShape({input_batch, input_height, input_width, input_depth})); in TEST_F() 160 const int input_batch = 1; in TEST_F() local 165 {input_batch, input_height, input_width, input_depth}); in TEST_F() 219 TensorShape({input_batch, input_height, input_width, input_depth})); in TEST_F()
|
D | mkl_conv_ops.h | 126 int input_batch = static_cast<int>(input_batch_raw); in GetInputSizeInMklOrder() local 141 mkldnn_sizes[MklDnnDims::Dim_N] = input_batch; in GetInputSizeInMklOrder() 165 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = input_batch; in GetInputSizeInMklOrder()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | convolution_dimension_numbers_test.cc | 38 int64 input_batch, int64 input_feature, int64 input_first_spatial, in CreateConvDimensionNumbers() argument 44 dimension_numbers.set_input_batch_dimension(input_batch); in CreateConvDimensionNumbers()
|
/external/tensorflow/tensorflow/python/debug/cli/ |
D | tensor_format.py | 323 input_batch = True 326 input_batch = False 400 if input_batch:
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | convolution_group_converter.cc | 222 int64 input_batch = activation->shape().dimensions(input_batch_dimension); in HandleBatchGroupCount() local 225 TF_RET_CHECK(input_batch == batch_group_count); in HandleBatchGroupCount()
|
D | shape_inference.cc | 1699 const int64 input_batch = lhs.dimensions(dnums.input_batch_dimension()); in InferConvolveShape() local 1710 if (batch_group_count > 1 && input_batch % kernel_output_features != 0) { in InferConvolveShape() 1716 input_batch, kernel_output_features, batch_group_count, in InferConvolveShape() 1751 if (input_batch % batch_group_count > 0) { in InferConvolveShape() 1757 input_batch, batch_group_count, ShapeUtil::HumanString(lhs), in InferConvolveShape() 1783 dimensions[dnums.output_batch_dimension()] = input_batch / batch_group_count; in InferConvolveShape()
|
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | reference_ops.h | 305 const int input_batch = input_shape.Dims(0); in DepthToSpace() local 317 TFLITE_DCHECK_EQ(input_batch, output_batch); in DepthToSpace() 358 const int input_batch = input_shape.Dims(0); in SpaceToDepth() local 370 TFLITE_DCHECK_EQ(input_batch, output_batch); in SpaceToDepth() 372 for (int in_b = 0; in_b < input_batch; ++in_b) { in SpaceToDepth() 3107 int input_batch = out_b % input_batch_size; in SpaceToBatchND() local 3123 Offset(input1_shape, input_batch, in SpaceToBatchND()
|