Home
last modified time | relevance | path

Searched refs:MakePartialShape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dpartial_tensor_shape_test.cc252 TF_ASSERT_OK(PartialTensorShape::MakePartialShape(dims, 0, &shape)); in TEST()
260 TF_ASSERT_OK(PartialTensorShape::MakePartialShape(dims, 3, &shape)); in TEST()
272 PartialTensorShape::MakePartialShape(dims, 3, &shape).code()); in TEST()
Dtensor_shape.h442 static Status MakePartialShape(const T* dims, int n, in MakePartialShape() function
/external/tensorflow/tensorflow/core/common_runtime/
Dconstant_folding_test.cc474 TF_EXPECT_OK(PartialTensorShape::MakePartialShape(r0_dims, 2, &ps0)); in TEST_F()
477 TF_EXPECT_OK(PartialTensorShape::MakePartialShape<int>(r1_dims, 3, &ps1)); in TEST_F()
553 TF_EXPECT_OK(PartialTensorShape::MakePartialShape(r0_dims, 2, &ps0)); in TEST_F()
609 TF_EXPECT_OK(PartialTensorShape::MakePartialShape(c0_dims, 0, &ps0)); in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc39 return PartialTensorShape::MakePartialShape(dims.data(), dims.size(), shape); in ShapeHandleToTensorShape()
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.cc157 return PartialTensorShape::MakePartialShape(t.vec<int32>().data(), in TensorShapeFromTensor()
160 return PartialTensorShape::MakePartialShape(t.vec<int64>().data(), in TensorShapeFromTensor()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc53 OP_REQUIRES_OK(ctx, PartialTensorShape::MakePartialShape( in MakeDataset()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc208 return PartialTensorShape::MakePartialShape(dims.data(), dims.size(), shape); in ShapeHandleToTensorShape()
/external/tensorflow/tensorflow/core/kernels/data/
Dpadded_batch_dataset_op.cc65 OP_REQUIRES_OK(ctx, PartialTensorShape::MakePartialShape( in MakeDataset()
/external/tensorflow/tensorflow/core/ops/
Darray_ops_test.cc967 TF_ASSERT_OK(PartialTensorShape::MakePartialShape(dims, 2, &shape)); in TEST()