| /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/compiler/xla/service/ |
| D | space_to_batch_converter_test.cc | 59 const int64 batch_dim = reshape->operand(0) in TEST_F() local
108 const int64 batch_dim = reshape->operand(0) in TEST_F() local
142 const int64 batch_dim = reshape->operand(0) in TEST_F() local
|
| D | batch_dot_simplification.cc | 57 for (int64 batch_dim : dim_numbers.lhs_batch_dimensions()) { in ElideDegenerateBatchDimensionFromBatchDot() local
|
| D | topk_rewriter.cc | 92 const int64 batch_dim = sort_dim == 1 ? 0 : 1; in Run() local
|
| D | dynamic_dimension_inference_test.cc | 944 WindowDimension* batch_dim = window.add_dimensions(); in TEST_F() local
994 WindowDimension* batch_dim = window.add_dimensions(); in TEST_F() local
|
| D | space_to_batch_converter.cc | 1135 const int64 batch_dim = result.first; in SupportedOpForPropagation() local
1266 const int64 batch_dim = DimLookUp(permute_dims, old_batch_dim); in Propagate() local
1727 const int64 batch_dim = DimLookUp(permute_dims, old_batch_dim); in BatchToSpace() 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>(OpKernelContext* context, int batch_dim, in CheckErrors()
|
| D | reverse_sequence_op.h | 32 ReverseGenerator(typename TTypes<T, Dims>::ConstTensor input, int32 batch_dim, in ReverseGenerator()
|
| D | conv_grad_shape_utils.cc | 114 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2() local
|
| D | scatter_nd_op.cc | 781 const int64 batch_dim = (indices.dims() > 1) ? indices.dims() - 1 : 1; in ValidateUpdateShape() 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
210 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 | 44 const int64 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 __anon30b5031d1402() local
857 DimensionHandle batch_dim; in __anon30b5031d1502() local
|
| D | boosted_trees_ops.cc | 495 shape_inference::DimensionHandle batch_dim = c->Dim(c->input(1), 0); in __anon770f674c0d02() local
|
| D | array_ops_test.cc | 1146 auto rebuild_node_def = [&op](const int32 seq_dim, const int32 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/gpu/ |
| D | gpu_layout_assignment.cc | 223 for (int64 batch_dim : dim_nums.lhs_batch_dimensions()) { in AddBackendConstraints() local
|
| D | gemm_thunk.cc | 197 for (int64 batch_dim : dim_nums.lhs_batch_dimensions()) { in RunGemm() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 285 for (int64 batch_dim = 0; batch_dim < batch_dims; ++batch_dim) { in TorchIndexSelect() local
|
| D | svd_test.cc | 64 Array3D<float> GetUnitMatrix3D(int32 batch_dim, int32 mat_dim) { in GetUnitMatrix3D()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 255 int batch_dim; in ReverseSequenceGrad() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 819 const uint64_t batch_dim = 0; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 520 DimensionHandle* batch_dim, in DimensionsFromShape()
544 Status ShapeFromDimensions(DimensionHandle batch_dim, in ShapeFromDimensions()
|