Searched refs:MakePartialShape (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | partial_tensor_shape_test.cc | 265 TF_ASSERT_OK(PartialTensorShape::MakePartialShape(dims, 0, &shape)); in TEST()
273 TF_ASSERT_OK(PartialTensorShape::MakePartialShape(dims, 3, &shape)); in TEST()
285 PartialTensorShape::MakePartialShape(dims, 3, &shape).code()); in TEST()
|
| D | tensor_shape.h | 557 static Status MakePartialShape(const T* dims, int n, in MakePartialShape() function
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device.cc | 27 return PartialTensorShape::MakePartialShape(dims.data(), num_dims, shape); in Shape()
|
| D | tensor_handle.cc | 588 auto s = PartialTensorShape::MakePartialShape(dims.data(), num_dims, in SetInferenceShape()
|
| /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/c/eager/ |
| D | unified_api_test.cc | 174 Status s = PartialTensorShape::MakePartialShape(dim_sizes, 2, &shape); in TEST_P()
|
| D | c_api_unified_experimental.cc | 151 Status status = tensorflow::PartialTensorShape::MakePartialShape( in TF_AddFunctionParameter()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.cc | 54 return PartialTensorShape::MakePartialShape(t.vec<int32>().data(), in TensorShapeFromTensor()
57 return PartialTensorShape::MakePartialShape(t.vec<int64>().data(), in TensorShapeFromTensor()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 42 return PartialTensorShape::MakePartialShape(dims.data(), dims.size(), shape); in ShapeHandleToTensorShape()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 51 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 | 429 OP_REQUIRES_OK(ctx, PartialTensorShape::MakePartialShape( in MakeDataset()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 365 TF_RETURN_IF_ERROR(PartialTensorShape::MakePartialShape( in CompileTFFunctionToHlo()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops_test.cc | 1002 TF_ASSERT_OK(PartialTensorShape::MakePartialShape(dims, 2, &shape)); in TEST()
|