Home
last modified time | relevance | path

Searched refs:input_batch (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_fft_impl.h36 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 …]
Druntime_conv2d_impl.h29 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()
Druntime_single_threaded_conv2d.cc27 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()
Druntime_conv2d_mkl.cc53 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()
Druntime_conv2d.cc29 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()
Druntime_single_threaded_fft.cc27 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()
Druntime_fft.cc30 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()
Druntime_single_threaded_conv2d.h27 tensorflow::int64 input_batch, tensorflow::int64 input_rows,
40 float* lhs, float* rhs, tensorflow::int64 input_batch,
Druntime_conv2d.h26 float* lhs, float* rhs, tensorflow::int64 input_batch,
41 tensorflow::int64 input_batch, tensorflow::int64 input_rows,
Druntime_fft.h26 tensorflow::int64 input_batch, tensorflow::int64 fft_length0,
Druntime_single_threaded_fft.h26 tensorflow::int64 input_batch, tensorflow::int64 fft_length0,
Druntime_conv2d_mkl.h26 float* lhs, float* rhs, tensorflow::int64 input_batch,
Dir_emitter.cc1125 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/
Dquantized_batch_norm_op_test.cc63 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()
Dmkl_conv_ops.h126 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/
Dconvolution_dimension_numbers_test.cc38 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/
Dtensor_format.py323 input_batch = True
326 input_batch = False
400 if input_batch:
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc222 int64 input_batch = activation->shape().dimensions(input_batch_dimension); in HandleBatchGroupCount() local
225 TF_RET_CHECK(input_batch == batch_group_count); in HandleBatchGroupCount()
Dshape_inference.cc1699 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/
Dreference_ops.h305 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()