Home
last modified time | relevance | path

Searched refs:GetDimension (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dmkl_input_conversion_op.cc174 for (size_t i = 0; i < mkl_shape->GetDimension(); ++i) in Compute()
193 mkl_output_mkl_shape.SetTfLayout(mkl_shape->GetDimension(), in Compute()
196 mkl_output_mkl_shape.SetTfDimOrder(mkl_shape->GetDimension()); in Compute()
206 tf_mkl_shape->SetTfLayout(mkl_shape->GetDimension(), in Compute()
319 mkl_output_mkl_shape.SetTfLayout(input_shape_0.GetDimension(), in Compute()
407 for (size_t i = 0; i < mkl_shape->GetDimension(); ++i) in Compute()
427 mkl_output_mkl_shape.SetTfLayout(mkl_shape->GetDimension(), in Compute()
Dmkl_pooling_ops_common.cc68 depth = mklInputShape->GetDimension('C'); in Init()
69 tensor_in_cols = mklInputShape->GetDimension('W'); in Init()
70 tensor_in_rows = mklInputShape->GetDimension('H'); in Init()
71 tensor_in_batch = mklInputShape->GetDimension('N'); in Init()
Dmkl_lrn_op.cc103 ? mkl_context.input_shape.GetDimension() in Compute()
383 ? mkl_context.inimage_shape.GetDimension() in Compute()
418 context, mkl_context.ingrad_shape.GetDimension() == 4, in Compute()
428 context, mkl_context.outimage_shape.GetDimension() == 4, in Compute()
779 } else if (!src_dnn_shape.IsMklChannelDim(src_dnn_shape.GetDimension() - in Compute()
964 OP_REQUIRES(context, src_dnn_shape.GetDimension() == 4, in SanityCheckInputs()
1032 input_grad_dnn_shape.GetDimension() - 1) && in Compute()
1034 orig_input_dnn_shape.GetDimension() - 1) && in Compute()
1036 orig_output_dnn_shape.GetDimension() - 1); in Compute()
1290 OP_REQUIRES(context, in_grads_dnn_shape.GetDimension() == 4, in SanityCheckInputs()
[all …]
Dmkl_concat_op.cc290 if (s.GetDimension() != 4) { in Compute()
329 size_t exp_dims = inpshape0.GetDimension(); in Compute()
330 OP_REQUIRES(context, s.GetDimension() == exp_dims, in Compute()
334 s.GetDimension(), " vs. expected rank = ", exp_dims)); in Compute()
356 s.GetDimension() > 0 ? s.GetSizes()[concat_dim] : 1; in Compute()
520 SIZE_OF_MKL_SERIAL_DATA(mkl_tensor_mkl_shape.GetDimension())); in CallEigenVersion()
550 SIZE_OF_MKL_SERIAL_DATA(mkl_tensor_mkl_shape.GetDimension())); in CallEigenVersion()
Dmkl_aggregate_ops.cc88 ? mkl_context.input1_shape.GetDimension() in Compute()
91 ? mkl_context.input2_shape.GetDimension() in Compute()
314 int src1_dims_size = input1_in_mkl_format ? src1_mkl_shape.GetDimension() in Compute()
316 int src2_dims_size = input2_in_mkl_format ? src2_mkl_shape.GetDimension() in Compute()
Dmkl_avgpooling_op.cc77 mkl_context.params.in_dim = mkl_context.input_shape.GetDimension(); in Compute()
264 mkl_context.params.in_dim = mkl_context.out_backprop_shape.GetDimension(); in Compute()
373 OP_REQUIRES(context, out_backprop_shape.GetDimension() == 4, in MklCreateLayoutsAndPrimitives()
683 original_input_mkl_shape.GetDimension() == 1 && in SanityCheckInputs()
695 OP_REQUIRES(context, input_gradient_mkl_shape.GetDimension() == 4, in SanityCheckInputs()
Dmkl_conv_grad_filter_ops.cc108 OP_REQUIRES(context, mkl_context.input_shape.GetDimension() == 4, in Compute()
121 context, mkl_context.out_backprop_shape.GetDimension() == 4, in Compute()
154 ? mkl_context.input_shape.GetDimension() in Compute()
157 ? mkl_context.out_backprop_shape.GetDimension() in Compute()
Dmkl_fused_batch_norm_op.cc245 ? mkl_shape_input_shape.GetDimension() in MklExtractParams()
528 ? mkl_shape_input_shape.GetDimension() in MklExtractParams()
724 OP_REQUIRES(context, dnn_shape_src.GetDimension() == 4, in Compute()
875 dnn_shape_dst.SetTfLayout(dnn_shape_src.GetDimension(), src_dims, in Compute()
1069 context, dnn_shape_diff_dst.GetDimension() == 4, in Compute()
1082 OP_REQUIRES(context, dnn_shape_src.GetDimension() == 4, in Compute()
1219 dnn_shape_diff_src.SetTfLayout(dnn_shape_src.GetDimension(), src_dims, in Compute()
1221 dnn_shape_diff_src.SetTfDimOrder(dnn_shape_src.GetDimension(), in Compute()
Dmkl_conv_grad_input_ops.cc107 OP_REQUIRES(context, mkl_context.filter_shape.GetDimension() == 4, in Compute()
113 const int64 filter_dims = mkl_context.filter_shape.GetDimension(); in Compute()
123 OP_REQUIRES(context, mkl_context.outback_shape.GetDimension() == 4, in Compute()
Dmkl_maxpooling_op.cc247 mkl_context.params.in_dim = mkl_context.input_shape.GetDimension(); in Compute()
755 OP_REQUIRES(context, orig_input_mkl_shape.GetDimension() == 4, in SanityCheckInputs()
764 OP_REQUIRES(context, orig_output_mkl_shape.GetDimension() == 4, in SanityCheckInputs()
772 OP_REQUIRES(context, grad_mkl_shape.GetDimension() == 4, in SanityCheckInputs()
Dmkl_relu_op.cc92 mkl_context.in_dims = mkl_context.input_shape.GetDimension(); in Compute()
302 mkl_context.in_dims = tmp_mkl_shape->GetDimension(); in Compute()
430 dnn_shape_dst.SetTfLayout(dnn_shape_src.GetDimension(), in Compute()
587 dnn_shape_diff_src.SetTfLayout(dnn_shape_src.GetDimension(), in Compute()
Dmkl_tfconv_op.h153 size_t ndims = input_shape.GetDimension(); in ConvertMklToTf()
Dmkl_conv_grad_bias_ops.cc69 context, mkl_context.input_shape.GetDimension() == 4, in Compute()
Dmkl_reshape_op.cc100 for (size_t i = 0; i < mkl_shape_input.GetDimension(); i++) { in Compute()
Dmkl_pooling_ops_common.h259 OP_REQUIRES(context, input_mkl_shape.GetDimension() == 4, in SanityCheckInput()
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dcholesky.cc48 const int64 n = xla::ShapeUtil::GetDimension(*shape, -2); in CholeskyUnblocked()
111 const int64 n = xla::ShapeUtil::GetDimension(*a_shape, -1); in Cholesky()
112 if (n != xla::ShapeUtil::GetDimension(*a_shape, -2)) { in Cholesky()
Dtriangular_solve.cc65 if (xla::ShapeUtil::GetDimension(*a_shape, -1) != in TriangularSolve()
66 xla::ShapeUtil::GetDimension(*a_shape, -2)) { in TriangularSolve()
71 const int64 m = xla::ShapeUtil::GetDimension(*b_shape, -2); in TriangularSolve()
72 const int64 n = xla::ShapeUtil::GetDimension(*b_shape, -1); in TriangularSolve()
73 if ((left_side ? m : n) != xla::ShapeUtil::GetDimension(*a_shape, -1)) { in TriangularSolve()
373 const int64 m = xla::ShapeUtil::GetDimension(*b_shape, -2); in TriangularSolveLeftLooking()
374 const int64 n = xla::ShapeUtil::GetDimension(*b_shape, -1); in TriangularSolveLeftLooking()
Dscatter.cc81 if (xla::ShapeUtil::GetDimension(*buffer_shape, i) == 0) { in XlaScatter()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc288 ShapeUtil::GetDimension(base_shape, i), dim.base_dilation()); in InferWindowOutputShape()
1263 if (ShapeUtil::GetDimension(offset_shape, 0) != feature_count) { in InferBatchNormTrainingShape()
1268 ShapeUtil::GetDimension(offset_shape, 0), feature_count); in InferBatchNormTrainingShape()
1271 if (ShapeUtil::GetDimension(scale_shape, 0) != feature_count) { in InferBatchNormTrainingShape()
1276 ShapeUtil::GetDimension(scale_shape, 0), feature_count); in InferBatchNormTrainingShape()
1397 if (ShapeUtil::GetDimension(offset_shape, 0) != feature_count) { in InferBatchNormInferenceShape()
1402 ShapeUtil::GetDimension(offset_shape, 0), feature_count); in InferBatchNormInferenceShape()
1405 if (ShapeUtil::GetDimension(scale_shape, 0) != feature_count) { in InferBatchNormInferenceShape()
1410 ShapeUtil::GetDimension(scale_shape, 0), feature_count); in InferBatchNormInferenceShape()
1413 if (ShapeUtil::GetDimension(mean_shape, 0) != feature_count) { in InferBatchNormInferenceShape()
[all …]
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h156 size_t GetDimension() const { return dimension_; } in GetDimension() function
426 inline size_t GetDimension(char dimension) const { in GetDimension() function
428 CHECK(index >= 0 && index < this->GetDimension()) in GetDimension()
449 inline size_t GetDimension() const { return data_.dimension_; } in GetDimension() function
684 for (size_t j = 0; j < mkl_shape.GetDimension(); j++) { in ConvertMklToTF()
808 second_shape.AddDim(SIZE_OF_MKL_SERIAL_DATA(mkl_shape.GetDimension())); in AllocateOutputSetMklShape()
842 second_shape.AddDim(SIZE_OF_MKL_SERIAL_DATA(mkl_shape.GetDimension())); in AllocateOutputSetMklShape()
931 size_t tf_dim = mkl_shape.GetDimension(); in MklSizesToTFSizes()
971 CHECK(index >= 0 && index < mkl_shape.GetDimension()) in GetMklTensorDim()
1144 if (input_shape_0->GetDimension() != input_shape_1->GetDimension()) { in MklCompareShapes()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc43 block_rank == xla::ShapeUtil::GetDimension(paddings.shape(), 0) && in SpaceToBatch()
44 2 == xla::ShapeUtil::GetDimension(paddings.shape(), 1), in SpaceToBatch()
Dbatchtospace_op.cc43 block_rank == xla::ShapeUtil::GetDimension(crops.shape(), 0) && in BatchToSpace()
44 2 == xla::ShapeUtil::GetDimension(crops.shape(), 1), in BatchToSpace()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc179 const int32 upper_bound = ShapeUtil::GetDimension(input_shape, i) - in MakeRandomNonwrappingSliceIndex()
180 ShapeUtil::GetDimension(slice_shape, i); in MakeRandomNonwrappingSliceIndex()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc33 EXPECT_EQ(3, ShapeUtil::GetDimension(matrix, -1)); in TEST()
34 EXPECT_EQ(2, ShapeUtil::GetDimension(matrix, -2)); in TEST()
39 ASSERT_EQ(4, ShapeUtil::GetDimension(shape, -1)); in TEST()
44 ASSERT_DEATH(ShapeUtil::GetDimension(matrix, -3), "dimension_number >= 0"); in TEST()
/external/tensorflow/tensorflow/compiler/xla/python/
Dnumpy_bridge.cc131 LongToPyIntOrPyLong(ShapeUtil::GetDimension(shape, i))); in PyShapeInfoFromXlaShape()
363 dimensions[i] = ShapeUtil::GetDimension(literal.shape(), i); in PyObjectFromXlaLiteral()

12