| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_impl.h | 36 int64 input_batch, int64 fft_length0, int64 fft_length1, in EigenFftC2C()
63 int64 input_batch, int64 fft_length0, int64 fft_length1, in EigenFftR2C()
102 int64 input_batch, int64 fft_length0, int64 fft_length1, in EigenFftC2R()
173 int32 fft_type, int64 input_batch, int64 fft_length0, in EigenFftWithRank()
208 int32 fft_type, int32 fft_rank, int64 input_batch, in EigenFftImpl()
|
| D | runtime_single_threaded_fft.cc | 27 int32 fft_rank, int64 input_batch, int64 fft_length0, int64 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()
|
| D | runtime_single_threaded_conv2d.cc | 27 Eigen::half* rhs, int64 input_batch, int64 input_rows, int64 input_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
45 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| D | runtime_conv2d.cc | 29 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_EigenConvF32()
47 Eigen::half* rhs, int64 input_batch, int64 input_rows, int64 input_cols, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_conv2d_mkl.cc | 53 ScalarType* rhs, int64 input_batch, int64 input_rows, in MKLConvImpl()
154 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_conv2d_impl.h | 29 ScalarType* rhs, int64 input_batch, int64 input_rows, in EigenConvImpl()
|
| D | ir_emitter.cc | 1125 int64 input_batch = input_shape.dimensions(dnums.input_batch_dimension()); in HandleConvolution() local
1270 int64 input_batch = 1; in HandleFft() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_batch_norm_op_test.cc | 63 const int input_batch = 1; in TEST_F() local
160 const int input_batch = 1; in TEST_F() local
|
| D | mkl_conv_ops.h | 126 int input_batch = static_cast<int>(input_batch_raw); in GetInputSizeInMklOrder() 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/compiler/xla/service/ |
| D | convolution_group_converter.cc | 222 int64 input_batch = activation->shape().dimensions(input_batch_dimension); in HandleBatchGroupCount() local
|
| D | shape_inference.cc | 1699 const int64 input_batch = lhs.dimensions(dnums.input_batch_dimension()); in InferConvolveShape() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 305 const int input_batch = input_shape.Dims(0); in DepthToSpace() local
358 const int input_batch = input_shape.Dims(0); in SpaceToDepth() local
3107 int input_batch = out_b % input_batch_size; in SpaceToBatchND() local
|