Searched refs:MakeShapeFromTensorShape (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/ |
| D | trt_shfn.cc | 49 c->MakeShapeFromTensorShape(input_shapes.at(i), &handle)); in TRTEngineOpShapeInference()
69 c->MakeShapeFromTensorShape(output_shapes.at(i), &handle)); in TRTEngineOpShapeInference()
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/ops/ |
| D | array_ops.cc | 59 TF_RETURN_IF_ERROR(c->MakeShapeFromTensorShape( in __anon7c7f74300102()
150 TF_RETURN_IF_ERROR(c->MakeShapeFromTensorShape(original_shape, &s)); in __anon7c7f74300202()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | io_ops.cc | 123 c->MakeShapeFromTensorShape(parsed_slice_shape, &shape_handle)); in __anonec23b74b0302()
237 c->MakeShapeFromTensorShape(parsed_slice_shape, &shape_handle)); in __anonec23b74b0802()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 296 TF_RETURN_IF_ERROR(c->MakeShapeFromTensorShape(shape_attr, &s)); in __anon3500048a0202()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 1145 TEST_F(ShapeInferenceTest, MakeShapeFromTensorShape) { in TEST_F() argument
1151 TF_ASSERT_OK(c.MakeShapeFromTensorShape(TensorShape(), &out)); in TEST_F()
1153 TF_ASSERT_OK(c.MakeShapeFromTensorShape(TensorShape({0}), &out)); in TEST_F()
1155 TF_ASSERT_OK(c.MakeShapeFromTensorShape(TensorShape({0, 7, 1000}), &out)); in TEST_F()
|
| D | shape_inference.h | 496 Status MakeShapeFromTensorShape(const TensorShape& shape, ShapeHandle* out);
|
| D | shape_inference.cc | 934 Status InferenceContext::MakeShapeFromTensorShape(const TensorShape& shape, in MakeShapeFromTensorShape() function in tensorflow::shape_inference::InferenceContext
|
| /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 | 459 status = context->MakeShapeFromTensorShape( in PropagateShapeInference()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1300 ic->MakeShapeFromTensorShape(t->shape(), &output_shape)); in UpdateOutputShapesAndValues()
|