| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_impl.h | 44 int64_t input_batch, int64_t fft_length0, int64_t fft_length1, in EigenFftC2C()
71 int64_t input_batch, int64_t fft_length0, int64_t fft_length1, in EigenFftR2C()
110 int64_t input_batch, int64_t fft_length0, int64_t fft_length1, in EigenFftC2R()
182 int64_t input_batch, int64_t fft_length0, in EigenFftWithRank()
246 int64_t input_batch, int64_t fft_length0, int64_t fft_length1, in EigenFftImpl()
|
| D | runtime_single_threaded_conv2d.cc | 25 Eigen::half* rhs, int64_t input_batch, int64_t input_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
44 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| D | runtime_conv2d.cc | 28 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_EigenConvF32()
48 Eigen::half* rhs, int64_t input_batch, int64_t input_rows, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_single_threaded_fft.cc | 27 int32_t double_precision, int32_t fft_rank, int64_t input_batch, in __xla_cpu_runtime_EigenSingleThreadedFft()
|
| D | runtime_fft.cc | 31 int32_t double_precision, int32_t fft_rank, int64_t 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_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_conv2d_impl.h | 33 ScalarType* rhs, Eigen::Index input_batch, in EigenConvImpl()
|
| D | ir_emitter.cc | 903 int64_t input_batch = in HandleConvolution() local
1037 int64_t 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 | 41 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_t input_batch, int64_t input_feature, int64_t 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/core/kernels/batching_util/ |
| D | shared_batch_scheduler.h | 757 auto input_batch = std::make_shared<BatchInputTask<TaskType>>( in ScheduleWithLazySplit() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 228 const int64_t input_batch = in HandleBatchGroupCount() local
|
| D | shape_inference.cc | 1731 const int64_t 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
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference.cc | 403 const int32_t input_batch = out_b % input_batch_size; in SpaceToBatchNDRank4() local
|
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 402 const int32_t input_batch = out_b % input_batch_size; in SpaceToBatchNDRank4() local
|
| D | onedevice_train_palm_detection_filter_inference_v2.cc | 402 const int32_t input_batch = out_b % input_batch_size; in SpaceToBatchNDRank4() local
|