| /external/tensorflow/tensorflow/core/framework/ |
| D | partial_tensor_shape_test.cc | 127 EXPECT_TRUE(a.AsTensorShape(&full)); in TEST()
129 EXPECT_TRUE(b.AsTensorShape(&full)); in TEST()
133 EXPECT_FALSE(c.AsTensorShape(&full)); in TEST()
134 EXPECT_FALSE(d.AsTensorShape(&full)); in TEST()
|
| D | tensor_shape.h | 551 bool AsTensorShape(TensorShape* shape) const;
|
| D | tensor_shape.cc | 926 bool PartialTensorShape::AsTensorShape(TensorShape* shape) const { in AsTensorShape() function in tensorflow::PartialTensorShape
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 435 if (!partial_processing_shape.AsTensorShape(processing_shape) || in ValidateStridedSliceOp()
436 !partial_final_shape.AsTensorShape(final_shape)) { in ValidateStridedSliceOp()
|
| D | example_proto_helper.cc | 663 dense_shapes[i].AsTensorShape(&dense_shape); in GetDenseShapes()
|
| D | example_proto_fast_parsing.cc | 1483 if (!config.dense[d].shape.AsTensorShape(&values_shape)) { in FastParseSingleExample()
2289 if (!c.shape.AsTensorShape(&example_shape) || in ParseContextDenseFeatures()
2496 if (!c.shape.AsTensorShape(&row_shape) || in ParseSequenceDenseFeatures()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 109 OP_REQUIRES(c, partial_element_shape.AsTensorShape(&element_shape), in Compute()
206 c, partial_element_shape.AsTensorShape(&element_shape), in Compute()
254 c, partial_element_shape.AsTensorShape(&element_shape), in Compute()
374 OP_REQUIRES(c, element_shape_except_first_dim.AsTensorShape(&output_shape), in Compute()
591 c, partial_element_shape.AsTensorShape(&element_shape), in Compute()
|
| D | tensor_array.h | 581 element_shape_.AsTensorShape(&element_shape); // Always succeeds. in LockedRead()
|
| D | tensor_array_ops.cc | 646 element_shape_.AsTensorShape(&empty_shape); in Compute()
850 element_shape_except0_.AsTensorShape(&empty_shape); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 84 ctx, partial_final_shape.AsTensorShape(&final_shape), in EmitDynamicSlice()
97 ctx, partial_processing_shape.AsTensorShape(&processing_shape), in EmitDynamicSlice()
247 ctx, partial_final_shape.AsTensorShape(&final_shape), in Compile()
|
| D | tensor_array_ops.cc | 161 CHECK(element_shape_.AsTensorShape(&shape)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape.cc | 89 if (!partial_tensor_shape.AsTensorShape(&fully_defined_tensor_shape)) { in TypeToShape()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference_test.cc | 169 EXPECT_TRUE(iter->second.at(0).handle_shape.AsTensorShape(&resource_shape)); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 299 output_shapes_.at(i).AsTensorShape(&output_shape); in SetupOutputs()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 459 TF_RET_CHECK(static_shape.AsTensorShape(&shape)); in ComputeArgumentShapes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 747 if (!partial_shape.AsTensorShape(&shape)) { in OptimizeGraph()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 535 bool fill = inference_shape_.AsTensorShape(shape); in Shape()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parse_example_dataset_op.cc | 110 dense_shapes_[i].AsTensorShape(&dense_shape); in ParseExampleDatasetOp()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 2704 if (!raw_var_shape.handle_shape.AsTensorShape(&var_shape) && in CreatePerHostDummyArgs()
2705 !raw_var_shape.shape.AsTensorShape(&var_shape)) { in CreatePerHostDummyArgs()
|