Home
last modified time | relevance | path

Searched refs:shape_handle (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dfunction_ops.cc45 shape_inference::ShapeHandle shape_handle; in __anon5bd79b600102() local
47 context->MakeShapeFromShapeProto(shape_proto, &shape_handle)); in __anon5bd79b600102()
48 context->set_output(0, shape_handle); in __anon5bd79b600102()
79 shape_inference::ShapeHandle shape_handle; in __anon5bd79b600102() local
81 context->MakeShapeFromShapeProto(shape_proto, &shape_handle)); in __anon5bd79b600102()
83 0, std::vector<shape_inference::ShapeAndType>{{shape_handle, dtype}}); in __anon5bd79b600102()
Dboosted_trees_ops.cc579 shape_inference::ShapeHandle shape_handle; in __anon770f674c1002() local
587 c->Merge(c->input(1), c->Vector(num_features), &shape_handle)); in __anon770f674c1002()
591 TF_RETURN_IF_ERROR(c->WithRank(c->input(i + 2), 1, &shape_handle)); in __anon770f674c1002()
592 auto shape_rank_1 = c->MakeShape({c->Dim(shape_handle, 0)}); in __anon770f674c1002()
593 auto shape_rank_2 = c->MakeShape({c->Dim(shape_handle, 0), 1}); in __anon770f674c1002()
597 c->WithRank(c->input(i + num_features + 2), 1, &shape_handle)); in __anon770f674c1002()
600 shape_rank_1, &shape_handle)); in __anon770f674c1002()
603 c->WithRank(c->input(i + num_features * 2 + 2), 1, &shape_handle)); in __anon770f674c1002()
605 shape_rank_1, &shape_handle)); in __anon770f674c1002()
608 c->WithRank(c->input(i + num_features * 3 + 2), 2, &shape_handle)); in __anon770f674c1002()
[all …]
Dtensor_forest_ops.cc67 shape_inference::ShapeHandle shape_handle; in __anon6270b1470302() local
70 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 2, &shape_handle)); in __anon6270b1470302()
72 batch_size = c->Dim(shape_handle, 0); in __anon6270b1470302()
Dio_ops.cc121 ShapeHandle shape_handle; in __anon52e493120302() local
123 c->MakeShapeFromTensorShape(parsed_slice_shape, &shape_handle)); in __anon52e493120302()
124 c->set_output(i, shape_handle); in __anon52e493120302()
235 ShapeHandle shape_handle; in __anon52e493120802() local
237 c->MakeShapeFromTensorShape(parsed_slice_shape, &shape_handle)); in __anon52e493120802()
238 c->set_output(0, shape_handle); in __anon52e493120802()
Dragged_conversion_ops.cc245 ShapeHandle shape_handle; in RaggedTensorToTensorShapeFn() local
247 c->MakeShapeFromShapeTensorTreatScalarAsUnknownShape(0, &shape_handle)); in RaggedTensorToTensorShapeFn()
248 c->ShapeHandleToProto(shape_handle, &shape); in RaggedTensorToTensorShapeFn()
/external/tensorflow/tensorflow/core/tpu/ops/
Dtpu_partitioned_input_op.cc69 ShapeHandle shape_handle = shapes_and_types->at(0).shape; in __anon2b8a3a590102() local
70 if (!c->FullyDefined(shape_handle)) { in __anon2b8a3a590102()
77 if (!c->Merge(shape_handle, previous_shape_handle, &tmp).ok()) { in __anon2b8a3a590102()
82 previous_shape_handle = shape_handle; in __anon2b8a3a590102()
/external/tensorflow/tensorflow/c/
Dops.cc228 TF_ShapeHandle* shape_handle, int64_t i, in TF_ShapeInferenceContextDim() argument
230 int64_t rank = TF_ShapeInferenceContextRank(ctx, shape_handle); in TF_ShapeInferenceContextDim()
239 auto* cc_shape_handle = reinterpret_cast<ShapeHandle*>(shape_handle); in TF_ShapeInferenceContextDim()
256 TF_ShapeHandle* shape_handle, in TF_ShapeInferenceContextSubshape() argument
263 Status s = cc_ctx->Subshape(*reinterpret_cast<ShapeHandle*>(shape_handle), in TF_ShapeInferenceContextSubshape()
Dops.h337 TF_ShapeInferenceContext* ctx, TF_ShapeHandle* shape_handle, int64_t i,
345 TF_ShapeInferenceContext* ctx, TF_ShapeHandle* shape_handle, int64_t start,
Dc_api_experimental.cc688 ShapeHandle shape_handle = c.output(i); in TFE_InferShapes() local
690 shape.num_dims = c.Rank(shape_handle); in TFE_InferShapes()
697 shape.dims[j] = c.Value(c.Dim(shape_handle, j)); in TFE_InferShapes()
Dc_api.cc299 tensorflow::shape_inference::ShapeHandle shape_handle = in TF_GraphSetOutputHandleShapesAndTypes() local
302 shape_handle, static_cast<DataType>(types[i])); in TF_GraphSetOutputHandleShapesAndTypes()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc472 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 …]
Dgraph_transferer.h191 const shape_inference::ShapeHandle& shape_handle,
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dshape_inference.cc51 shape_inference::ShapeHandle shape_handle = ic.output(i); in RunShapeInference() local
52 retvals[i]->SetInferenceShape(&ic, shape_handle); in RunShapeInference()
Dtensor_handle.cc546 shape_inference::ShapeHandle* shape_handle) { in InferenceShape() argument
557 *shape_handle = inference_context->MakeShape(dims_handle); in InferenceShape()
561 *shape_handle = inference_context->UnknownShape(); in InferenceShape()
569 *shape_handle = inference_context->MakeShape(dims_handle); in InferenceShape()
576 const shape_inference::ShapeHandle& shape_handle) { in SetInferenceShape() argument
577 auto num_dims = inference_context->Rank(shape_handle); in SetInferenceShape()
586 dims[i] = inference_context->Value(inference_context->Dim(shape_handle, i)); in SetInferenceShape()
Dtensor_handle.h234 shape_inference::ShapeHandle* shape_handle);
237 const shape_inference::ShapeHandle& shape_handle);
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc833 const auto& shape_handle = shape_and_type.shape; in UpdateFunction() local
836 shape_handle, shape_attr.mutable_list()->add_shape()); in UpdateFunction()
1490 const ShapeHandle& shape_handle = ic->output(i); in ShouldUpdateOutputShapesAndValues() local
1491 if (!ic->FullyDefined(shape_handle) || in ShouldUpdateOutputShapesAndValues()
1492 ic->Value(ic->NumElements(shape_handle)) > max_size) { in ShouldUpdateOutputShapesAndValues()
1514 const ShapeHandle& shape_handle = ic->input_tensors_as_shapes()[i]; in CreateInputTensors() local
1516 int32 rank = ic->Rank(shape_handle); in CreateInputTensors()
1526 int32 dim = ic->Value(ic->Dim(shape_handle, j)); in CreateInputTensors()
1532 int64 dim = ic->Value(ic->Dim(shape_handle, j)); in CreateInputTensors()
1757 const ShapeHandle& shape_handle = in MaybeUpdateNodeContextOutput() local
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dshape_inference_utils.cc306 ShapeHandle shape_handle = c.output(output); in InferReturnTypeComponentsForTFOp() local
308 << c.DebugString(shape_handle) << "\n"); in InferReturnTypeComponentsForTFOp()
334 CreateShapedTypeComponents(c, shape_handle, new_element_type)); in InferReturnTypeComponentsForTFOp()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc501 shape_inference::ShapeHandle shape_handle = context->output(output_index); in BuildTensorShapeMapFromGraph() local
502 if (context->RankKnown(shape_handle)) { in BuildTensorShapeMapFromGraph()
504 for (int k = 0; k < context->Rank(shape_handle); ++k) { in BuildTensorShapeMapFromGraph()
505 shape_inference::DimensionHandle dh = context->Dim(shape_handle, k); in BuildTensorShapeMapFromGraph()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc459 const ShapeHandle shape_handle, in CheckFormatConstraintsOnShape() argument
464 const int num_dims = c->Rank(shape_handle); in CheckFormatConstraintsOnShape()
466 shape_handle, GetTensorInnerFeatureDimIndex(num_dims, tensor_format)); in CheckFormatConstraintsOnShape()