Searched refs:AsTensorShape (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | partial_tensor_shape_test.cc | 114 EXPECT_TRUE(a.AsTensorShape(&full)); in TEST()
116 EXPECT_TRUE(b.AsTensorShape(&full)); in TEST()
120 EXPECT_FALSE(c.AsTensorShape(&full)); in TEST()
121 EXPECT_FALSE(d.AsTensorShape(&full)); in TEST()
|
| D | tensor_shape.h | 436 bool AsTensorShape(TensorShape* shape) const;
|
| D | tensor_shape.cc | 681 bool PartialTensorShape::AsTensorShape(TensorShape* shape) const { in AsTensorShape() function in tensorflow::PartialTensorShape
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.h | 190 dense_shapes[i].AsTensorShape(&dense_shape); in Init()
256 dense_shapes[i].AsTensorShape(&dense_shape); in Init()
|
| D | strided_slice_op.cc | 392 if (!partial_processing_shape.AsTensorShape(processing_shape) || in ValidateStridedSliceOp()
393 !partial_final_shape.AsTensorShape(final_shape)) { in ValidateStridedSliceOp()
|
| D | example_proto_fast_parsing.cc | 1272 if (!config.dense[d].shape.AsTensorShape(&values_shape)) { in FastParseSingleExample()
2007 if (!c.shape.AsTensorShape(&example_shape) || in FastParseSequenceExample()
2202 if (!c.shape.AsTensorShape(&row_shape) || in FastParseSequenceExample()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 130 OP_REQUIRES(c, partial_element_shape.AsTensorShape(&element_shape), in Compute()
227 c, partial_element_shape.AsTensorShape(&element_shape), in Compute()
275 c, partial_element_shape.AsTensorShape(&element_shape), in Compute()
399 c, element_shape_except_first_dim_.AsTensorShape(&output_shape), in Compute()
614 c, partial_element_shape.AsTensorShape(&element_shape), in Compute()
|
| D | tensor_array.h | 582 element_shape_.AsTensorShape(&element_shape); // Always succeeds. in LockedRead()
|
| D | tensor_array_ops.cc | 651 element_shape_.AsTensorShape(&empty_shape); in Compute()
856 element_shape_except0_.AsTensorShape(&empty_shape); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 309 output_shapes_.at(i).AsTensorShape(&output_shape); in SetupOutputs()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 80 partial_element_shape.AsTensorShape(&element_shape); in CreateZerosList()
|
| D | tensor_array_ops.cc | 161 CHECK(element_shape_.AsTensorShape(&shape)); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 66 dense_shapes_[i].AsTensorShape(&dense_shape); in ParseExampleDatasetOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 672 if (!partial_shape.AsTensorShape(&shape)) { in OptimizeGraph()
|