| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | utils_test.cc | 27 unsigned int batch_dim, height_dim, width_dim, depth_dim; in TEST() local
45 unsigned int batch_dim, height_dim, width_dim, depth_dim; in TEST() local
61 unsigned int batch_dim, height_dim, width_dim, depth_dim; in TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_sequence_op.cc | 44 void CheckErrors(OpKernelContext* context, int batch_dim, int seq_dim) { in CheckErrors()
79 void CheckErrorsGPU(OpKernelContext* context, int batch_dim, int seq_dim) { in CheckErrorsGPU()
100 void CheckErrors<GPUDevice, int32>(OpKernelContext* context, int batch_dim, in CheckErrors()
106 void CheckErrors<GPUDevice, int64_t>(OpKernelContext* context, int batch_dim, in CheckErrors()
|
| D | scatter_nd_util.cc | 29 const int64_t batch_dim = in ValidateScatterNdUpdateShape() local
|
| D | reverse_sequence_op.h | 32 typename TTypes<T, Dims>::ConstTensor input, int32_t batch_dim, in ReverseGenerator()
|
| D | conv_grad_shape_utils.cc | 115 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter_test.cc | 62 const int64_t batch_dim = previous_reshape->operand(0) in TEST_F() local
205 const int64_t batch_dim = previous_reshape->operand(0) in TEST_F() local
242 const int64_t batch_dim = previous_reshape->operand(0) in TEST_F() local
|
| D | topk_rewriter.cc | 139 const int64_t batch_dim = sort_dim == 1 ? 0 : 1; in SortIsInTopK() local
207 const int64_t batch_dim = sort_dim == 1 ? 0 : 1; in TransformToCustomCall() local
|
| D | batch_dot_simplification.cc | 56 for (int64_t batch_dim : dim_numbers.lhs_batch_dimensions()) { in ElideDegenerateBatchDimensionFromBatchDot() local
|
| D | dynamic_dimension_inference_test.cc | 951 WindowDimension* batch_dim = window.add_dimensions(); in TEST_F() local
1001 WindowDimension* batch_dim = window.add_dimensions(); in TEST_F() local
|
| D | space_to_batch_converter.cc | 1632 const int64_t batch_dim = result[DimMapper(SpaceToBatchDimMap::kBatch)]; in SupportedOpForPropagation() local
1803 const int64_t batch_dim = DimLookUp(permute_dims, old_batch_dim); in Propagate() local
2389 const int64_t batch_dim = DimLookUp(permute_dims, old_batch_dim); in BatchToSpace() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 106 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in CheckConvAttrs() local
209 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp() local
300 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp() local
|
| D | extract_image_patches_op.cc | 61 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in Compile() local
|
| D | scatter_nd_op.cc | 46 const int64_t batch_dim = indices_shape.dims() - 1; in ValidateUpdateShape() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops.cc | 30 Status SetOutputToSizedImage(InferenceContext* c, DimensionHandle batch_dim, in SetOutputToSizedImage()
228 DimensionHandle batch_dim = c->Dim(boxes, 0); in CombinedNMSShapeFn() local
822 DimensionHandle batch_dim; in __anonf6f833bb1402() local
857 DimensionHandle batch_dim; in __anonf6f833bb1502() local
|
| D | boosted_trees_ops.cc | 495 shape_inference::DimensionHandle batch_dim = c->Dim(c->input(1), 0); in __anon0a5055ea0d02() local
|
| D | array_ops_test.cc | 1157 const int32_t batch_dim) { in TEST()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reverse_sequence.cc | 83 int batch_dim = params->batch_dim; in ReverseSequenceImpl() local
|
| D | reverse_sequence_test.cc | 33 int seq_dim, int batch_dim) { in ReverseSequenceOpModel()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | custom_call_handler.cc | 82 const int64_t batch_dim = 0; in HandleCustomCallTopK() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/ |
| D | gemm_pattern.cc | 232 for (auto batch_dim : dim_nums.lhs_batching_dimensions()) { in GemmOpConversionRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd_test.cc | 64 Array3D<float> GetUnitMatrix3D(int32_t batch_dim, int32_t mat_dim) { in GetUnitMatrix3D()
|
| D | slicing.cc | 289 for (int64_t batch_dim = 0; batch_dim < batch_dims; ++batch_dim) { in TorchIndexSelect() local
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | Conv3D.cpp | 32 constexpr unsigned int batch_dim = 4u; variable
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 500 bool NeedsReformatTypeAndPermutation(int batch_dim, int feature_dim, in NeedsReformatTypeAndPermutation()
513 GetReformatTypeAndPermutation(int batch_dim, int feature_dim, in GetReformatTypeAndPermutation()
539 Value FormatToNHWC(Value value, int batch_dim, int feature_dim, in FormatToNHWC()
2318 const uint64_t batch_dim = 0; in IsSpatialPoolingWithoutDilation() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 269 int batch_dim; in ReverseSequenceGrad() local
|