Home
last modified time | relevance | path

Searched defs:shape_handle (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dtensor_forest_ops.cc67 shape_inference::ShapeHandle shape_handle; in __anoncf2fc2a90302() local
Dio_ops.cc121 ShapeHandle shape_handle; in __anone43acdb40302() local
235 ShapeHandle shape_handle; in __anone43acdb40802() local
Dboosted_trees_ops.cc301 shape_inference::ShapeHandle shape_handle; in __anone3ce78ae0b02() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc472 shape_inference::ShapeHandle shape_handle = context->output(output_index); in RegisterConstantNode() local
617 shape_inference::ShapeHandle shape_handle = context->output(0); in IsNodeFlattenReshape() local
964 const shape_inference::ShapeHandle& shape_handle, in BuildShapeArray()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1219 const ShapeHandle& shape_handle = ic->output(i); in ShouldUpdateOutputShapesAndValues() local
1243 const ShapeHandle& shape_handle = ic->input_tensors_as_shapes()[i]; in CreateInputTensors() local
1461 const ShapeHandle& shape_handle = ic->input_tensors_as_shapes()[i]; in MaybeUpdateNodeContextOutput() local
1974 ShapeHandle shape_handle; in UpdateQueue() local
/external/tensorflow/tensorflow/contrib/coder/kernels/
Drange_coder_ops_test.cc347 auto shape_handle = context->output(0); in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc500 shape_inference::ShapeHandle shape_handle = context->output(output_index); in BuildTensorShapeMapFromGraph() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc313 const ShapeHandle shape_handle, in CheckFormatConstraintsOnShape()
/external/tensorflow/tensorflow/c/
Dc_api.cc742 tensorflow::shape_inference::ShapeHandle shape_handle = in TF_GraphSetOutputHandleShapesAndTypes() local