| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_impl.h | 44 int64 input_batch, int64 fft_length0, int64 fft_length1, in EigenFftC2C()
71 int64 input_batch, int64 fft_length0, int64 fft_length1, in EigenFftR2C()
110 int64 input_batch, int64 fft_length0, int64 fft_length1, in EigenFftC2R()
182 int64 input_batch, int64 fft_length0, int64 fft_length1, in EigenFftWithRank()
246 int64 input_batch, int64 fft_length0, int64 fft_length1, in EigenFftImpl()
|
| D | runtime_single_threaded_conv2d.cc | 25 Eigen::half* rhs, tensorflow::int64 input_batch, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
47 tensorflow::int64 input_batch, tensorflow::int64 input_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| D | runtime_conv2d.cc | 28 tensorflow::int64 input_batch, tensorflow::int64 input_rows, in __xla_cpu_runtime_EigenConvF32()
51 Eigen::half* rhs, tensorflow::int64 input_batch, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_single_threaded_fft.cc | 27 int32 double_precision, int32 fft_rank, int64 input_batch, in __xla_cpu_runtime_EigenSingleThreadedFft()
|
| D | runtime_fft.cc | 31 int32 double_precision, int32 fft_rank, int64 input_batch, in __xla_cpu_runtime_EigenFft()
|
| D | runtime_conv2d_mkl.cc | 53 ScalarType* rhs, int64 input_batch, int64 input_rows, in MKLConvImpl()
156 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_conv2d_impl.h | 33 ScalarType* rhs, Eigen::Index input_batch, in EigenConvImpl()
|
| D | ir_emitter.cc | 902 int64 input_batch = input_shape.dimensions(dnums.input_batch_dimension()); in HandleConvolution() local
1035 int64 input_batch = 1; in HandleFft() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | depthwise_conv_test.cc | 475 const int input_batch = 2; in BatchPaddingValidTest() local
536 const int input_batch = 4; in BatchPaddingSameTest() local
1069 const int input_batch = 1; in TEST_P() local
1142 const int input_batch = 1; in TEST_P() local
1217 const int input_batch = 2; in TEST_P() local
1281 const int input_batch = 2; in TEST_P() local
1354 const int input_batch = 1; in TEST_P() local
1418 const int input_batch = 1; in TEST_P() local
1488 const int input_batch = 1; in TEST_P() local
1546 const int input_batch = 1; in TEST_P() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_batch_norm_op_test.cc | 62 const int input_batch = 1; in TEST_F() local
159 const int input_batch = 1; in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | space_to_depth.h | 39 const int input_batch = input_shape.Dims(0); in SpaceToDepth() local
|
| D | depth_to_space.h | 39 const int input_batch = input_shape.Dims(0); in DepthToSpace() local
|
| D | space_to_batch_nd.h | 80 int input_batch = out_b % input_batch_size; in SpaceToBatchND() local
|
| /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()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 132 int input_batch = static_cast<int>(input_batch_raw); in GetInputSizeInMklOrder() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 228 const int64 input_batch = in HandleBatchGroupCount() local
|
| D | shape_inference.cc | 1739 const int64 input_batch = lhs.dimensions(dnums.input_batch_dimension()); in InferConvolveShape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 286 int64_t input_batch = input_type.getDimSize(0); in Verify() local
348 int64_t input_batch = input_shape[0]; in Verify() local
|