| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_sequence_op.cc | 44 void CheckErrors(OpKernelContext* context, int batch_dim, int seq_dim) { in CheckErrors()
78 void CheckErrorsGPU(OpKernelContext* context, int batch_dim, int seq_dim) { in CheckErrorsGPU()
98 void CheckErrors<GPUDevice, int32>(OpKernelContext* context, int batch_dim, in CheckErrors()
104 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_ops.cc | 118 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2() local
|
| D | scatter_nd_op.cc | 522 const int64 batch_dim = (indices.dims() > 1) ? indices.dims() - 1 : 1; in ValidateUpdateShape() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batch_dot_simplification.cc | 39 for (int64 batch_dim : dim_numbers.lhs_batch_dimensions()) { in ElideDegenerateBatchDimensionFromBatchDot() local
|
| D | dynamic_dimension_inference_test.cc | 558 WindowDimension* batch_dim = window.add_dimensions(); in TEST_F() local
608 WindowDimension* batch_dim = window.add_dimensions(); in TEST_F() local
|
| /external/tensorflow/tensorflow/contrib/image/ops/ |
| D | image_ops.cc | 30 Status SetOutputToSizedImage(InferenceContext* c, DimensionHandle batch_dim, in SetOutputToSizedImage()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 173 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in CheckConvAttrs() local
269 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp() local
329 int batch_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp() local
|
| D | scatter_nd_op.cc | 42 const int64 batch_dim = indices_shape.dims() - 1; in ValidateUpdateShape() local
|
| D | extract_image_patches_op.cc | 54 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in Compile() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops.cc | 30 Status SetOutputToSizedImage(InferenceContext* c, DimensionHandle batch_dim, in SetOutputToSizedImage()
168 DimensionHandle batch_dim = c->Dim(boxes, 0); in CombinedNMSShapeFn() local
689 DimensionHandle batch_dim; in __anon6a71d27f1202() local
|
| D | array_ops_test.cc | 1111 auto rebuild_node_def = [&op](const int32 seq_dim, const int32 batch_dim) { in TEST()
|
| D | array_ops.cc | 1482 int64 batch_dim; in __anon7c94107b2402() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reverse_sequence.cc | 73 int batch_dim = params->batch_dim; in ReverseSequenceImpl() local
|
| D | reverse_sequence_test.cc | 30 int seq_dim, int batch_dim) { in ReverseSequenceOpModel()
|
| /external/tensorflow/tensorflow/contrib/gan/python/features/python/ |
| D | virtual_batchnorm_impl.py | 98 def _validate_call_input(tensor_list, batch_dim): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 200 for (int64 batch_dim : dim_nums.lhs_batch_dimensions()) { in AddBackendConstraints() local
|
| D | gemm_thunk.cc | 335 for (int64 batch_dim : dim_nums.lhs_batch_dimensions()) { in ExecuteOnStream() local
|
| D | ir_emitter.cc | 514 for (int64 batch_dim : dnums.lhs_batch_dimensions()) { in HandleDot() local
|
| /external/tensorflow/tensorflow/contrib/recurrent/python/ops/ |
| D | functional_rnn.py | 403 def _reverse(input_, seq_lengths, seq_dim, batch_dim): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd_test.cc | 59 Array3D<float> GetUnitMatrix3D(int32 batch_dim, int32 mat_dim) { in GetUnitMatrix3D()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 246 int batch_dim; in ReverseSequenceGrad() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 353 DimensionHandle* batch_dim, in DimensionsFromShape()
377 Status ShapeFromDimensions(DimensionHandle batch_dim, in ShapeFromDimensions()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 373 int batch_dim; member
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3248 batch_dim=None): argument
|