Searched refs:MakePartialShape (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | partial_tensor_shape_test.cc | 252 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()
|
D | tensor_shape.h | 442 static Status MakePartialShape(const T* dims, int n, in MakePartialShape() function
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | constant_folding_test.cc | 474 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/ |
D | shape_inference.cc | 39 return PartialTensorShape::MakePartialShape(dims.data(), dims.size(), shape); in ShapeHandleToTensorShape()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | list_kernels.cc | 157 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/ |
D | dense_to_sparse_batch_dataset_op.cc | 53 OP_REQUIRES_OK(ctx, PartialTensorShape::MakePartialShape( in MakeDataset()
|
/external/tensorflow/tensorflow/tools/graph_transforms/ |
D | fold_constants_lib.cc | 208 return PartialTensorShape::MakePartialShape(dims.data(), dims.size(), shape); in ShapeHandleToTensorShape()
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | padded_batch_dataset_op.cc | 65 OP_REQUIRES_OK(ctx, PartialTensorShape::MakePartialShape( in MakeDataset()
|
/external/tensorflow/tensorflow/core/ops/ |
D | array_ops_test.cc | 967 TF_ASSERT_OK(PartialTensorShape::MakePartialShape(dims, 2, &shape)); in TEST()
|