| /external/tensorflow/tensorflow/core/ops/ |
| D | boosted_trees_ops.cc | 301 shape_inference::ShapeHandle shape_handle; in __anone3ce78ae0b02() local
309 c->Merge(c->input(1), c->Vector(num_features), &shape_handle)); in __anone3ce78ae0b02()
313 TF_RETURN_IF_ERROR(c->WithRank(c->input(i + 2), 1, &shape_handle)); in __anone3ce78ae0b02()
314 auto shape_rank_1 = c->MakeShape({c->Dim(shape_handle, 0)}); in __anone3ce78ae0b02()
315 auto shape_rank_2 = c->MakeShape({c->Dim(shape_handle, 0), 1}); in __anone3ce78ae0b02()
319 c->WithRank(c->input(i + num_features + 2), 1, &shape_handle)); in __anone3ce78ae0b02()
322 shape_rank_1, &shape_handle)); in __anone3ce78ae0b02()
325 c->WithRank(c->input(i + num_features * 2 + 2), 1, &shape_handle)); in __anone3ce78ae0b02()
327 shape_rank_1, &shape_handle)); in __anone3ce78ae0b02()
330 c->WithRank(c->input(i + num_features * 3 + 2), 2, &shape_handle)); in __anone3ce78ae0b02()
[all …]
|
| D | tensor_forest_ops.cc | 67 shape_inference::ShapeHandle shape_handle; in __anoncf2fc2a90302() local
70 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &shape_handle)); in __anoncf2fc2a90302()
72 batch_size = c->Dim(shape_handle, 0); in __anoncf2fc2a90302()
|
| D | io_ops.cc | 121 ShapeHandle shape_handle; in __anone43acdb40302() local
123 c->MakeShapeFromTensorShape(parsed_slice_shape, &shape_handle)); in __anone43acdb40302()
124 c->set_output(i, shape_handle); in __anone43acdb40302()
235 ShapeHandle shape_handle; in __anone43acdb40802() local
237 c->MakeShapeFromTensorShape(parsed_slice_shape, &shape_handle)); in __anone43acdb40802()
238 c->set_output(0, shape_handle); in __anone43acdb40802()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 472 shape_inference::ShapeHandle shape_handle = context->output(output_index); in RegisterConstantNode() local
474 context->NumElements(shape_handle); in RegisterConstantNode()
481 shape_array = BuildShapeArray(shape_handle, context); in RegisterConstantNode()
617 shape_inference::ShapeHandle shape_handle = context->output(0); in IsNodeFlattenReshape() local
620 context->NumElements(shape_handle); in IsNodeFlattenReshape()
624 shape_array = BuildShapeArray(shape_handle, context); in IsNodeFlattenReshape()
964 const shape_inference::ShapeHandle& shape_handle, in BuildShapeArray() argument
966 switch (context->Rank(shape_handle)) { in BuildShapeArray()
971 {1, 1, 1, context->Value(context->Dim(shape_handle, 0))}}; in BuildShapeArray()
974 {1, 1, context->Value(context->Dim(shape_handle, 0)), in BuildShapeArray()
[all …]
|
| D | graph_transferer.h | 191 const shape_inference::ShapeHandle& shape_handle,
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | range_coder_ops_test.cc | 347 auto shape_handle = context->output(0); in TEST_F() local
349 ASSERT_EQ(context->Rank(shape_handle), 3); in TEST_F()
350 EXPECT_EQ(context->Value(context->Dim(shape_handle, 0)), 4); in TEST_F()
351 EXPECT_EQ(context->Value(context->Dim(shape_handle, 1)), 6); in TEST_F()
352 EXPECT_EQ(context->Value(context->Dim(shape_handle, 2)), 8); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1219 const ShapeHandle& shape_handle = ic->output(i); in ShouldUpdateOutputShapesAndValues() local
1220 if (!ic->FullyDefined(shape_handle) || in ShouldUpdateOutputShapesAndValues()
1221 ic->Value(ic->NumElements(shape_handle)) > max_size) { in ShouldUpdateOutputShapesAndValues()
1243 const ShapeHandle& shape_handle = ic->input_tensors_as_shapes()[i]; in CreateInputTensors() local
1245 int32 rank = ic->Rank(shape_handle); in CreateInputTensors()
1255 int32 dim = ic->Value(ic->Dim(shape_handle, j)); in CreateInputTensors()
1261 int64 dim = ic->Value(ic->Dim(shape_handle, j)); in CreateInputTensors()
1461 const ShapeHandle& shape_handle = ic->input_tensors_as_shapes()[i]; in MaybeUpdateNodeContextOutput() local
1462 if (ic->RankKnown(shape_handle) && ic->Rank(shape_handle) >= 1 && in MaybeUpdateNodeContextOutput()
1463 ic->ValueKnown(ic->Dim(shape_handle, 0))) { in MaybeUpdateNodeContextOutput()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 500 shape_inference::ShapeHandle shape_handle = context->output(output_index); in BuildTensorShapeMapFromGraph() local
501 if (context->RankKnown(shape_handle)) { in BuildTensorShapeMapFromGraph()
503 for (int k = 0; k < context->Rank(shape_handle); ++k) { in BuildTensorShapeMapFromGraph()
504 shape_inference::DimensionHandle dh = context->Dim(shape_handle, k); in BuildTensorShapeMapFromGraph()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 313 const ShapeHandle shape_handle, in CheckFormatConstraintsOnShape() argument
318 const int num_dims = c->Rank(shape_handle); in CheckFormatConstraintsOnShape()
320 shape_handle, GetTensorInnerFeatureDimIndex(num_dims, tensor_format)); in CheckFormatConstraintsOnShape()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 742 tensorflow::shape_inference::ShapeHandle shape_handle = in TF_GraphSetOutputHandleShapesAndTypes() local
745 shape_handle, static_cast<DataType>(types[i])); in TF_GraphSetOutputHandleShapesAndTypes()
|