Searched refs:MakeShapeFromTensorShape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | io_ops.cc | 123 c->MakeShapeFromTensorShape(parsed_slice_shape, &shape_handle)); in __anon52e493120302()
237 c->MakeShapeFromTensorShape(parsed_slice_shape, &shape_handle)); in __anon52e493120802()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 512 TF_RETURN_IF_ERROR(c->MakeShapeFromTensorShape(shape_attr, &s)); in __anonbbc1c2d10402()
887 TF_RETURN_IF_ERROR(c->MakeShapeFromTensorShape(shape_attr, &s)); in __anonbbc1c2d10d02()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 1155 TEST_F(ShapeInferenceTest, MakeShapeFromTensorShape) { in TEST_F() argument
1161 TF_ASSERT_OK(c.MakeShapeFromTensorShape(TensorShape(), &out)); in TEST_F()
1163 TF_ASSERT_OK(c.MakeShapeFromTensorShape(TensorShape({0}), &out)); in TEST_F()
1165 TF_ASSERT_OK(c.MakeShapeFromTensorShape(TensorShape({0, 7, 1000}), &out)); in TEST_F()
|
| D | shape_inference.h | 482 Status MakeShapeFromTensorShape(const TensorShape& shape, ShapeHandle* out);
|
| D | shape_inference.cc | 878 Status InferenceContext::MakeShapeFromTensorShape(const TensorShape& shape, in MakeShapeFromTensorShape() function in tensorflow::shape_inference::InferenceContext
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 331 shape_ctx->MakeShapeFromTensorShape(dtoh_shapes.at(i), &handle)); in InferOutputShapes()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 363 TF_RETURN_IF_ERROR(context->MakeShapeFromTensorShape( in TransformGraphToAddAggregatedInputNode()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 460 status = context->MakeShapeFromTensorShape( in PropagateShapeInference()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1571 ic->MakeShapeFromTensorShape(t->shape(), &output_shape)); in UpdateOutputShapesAndValues()
|