Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/ops/
Dfunction_ops.cc45 shape_inference::ShapeHandle shape_handle; in __anonfc9946590102() local
79 shape_inference::ShapeHandle shape_handle; in __anonfc9946590102() local
Dtensor_forest_ops.cc67 shape_inference::ShapeHandle shape_handle; in __anon75a11fe00302() local
Dio_ops.cc121 ShapeHandle shape_handle; in __anonec23b74b0302() local
235 ShapeHandle shape_handle; in __anonec23b74b0802() local
Dragged_conversion_ops.cc245 ShapeHandle shape_handle; in RaggedTensorToTensorShapeFn() local
Dboosted_trees_ops.cc579 shape_inference::ShapeHandle shape_handle; in __anon8a3fd5e51002() local
645 shape_inference::ShapeHandle shape_handle; in __anon8a3fd5e51102() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dshape_inference.cc51 shape_inference::ShapeHandle shape_handle = ic.output(i); in RunShapeInference() local
Dtensor_handle.cc546 shape_inference::ShapeHandle* shape_handle) { in InferenceShape()
576 const shape_inference::ShapeHandle& shape_handle) { in SetInferenceShape()
/external/tensorflow/tensorflow/core/tpu/ops/
Dtpu_partitioned_input_op.cc69 ShapeHandle shape_handle = shapes_and_types->at(0).shape; in __anonb44cb7520102() local
/external/tensorflow/tensorflow/c/
Dops.cc228 TF_ShapeHandle* shape_handle, int64_t i, in TF_ShapeInferenceContextDim()
256 TF_ShapeHandle* shape_handle, in TF_ShapeInferenceContextSubshape()
Dc_api_experimental.cc688 ShapeHandle shape_handle = c.output(i); in TFE_InferShapes() local
Dc_api.cc299 tensorflow::shape_inference::ShapeHandle shape_handle = in TF_GraphSetOutputHandleShapesAndTypes() 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/compiler/mlir/tensorflow/utils/
Dshape_inference_utils.cc306 ShapeHandle shape_handle = c.output(output); in InferReturnTypeComponentsForTFOp() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc833 const auto& shape_handle = shape_and_type.shape; in UpdateFunction() local
1490 const ShapeHandle& shape_handle = ic->output(i); in ShouldUpdateOutputShapesAndValues() local
1514 const ShapeHandle& shape_handle = ic->input_tensors_as_shapes()[i]; in CreateInputTensors() local
1757 const ShapeHandle& shape_handle = in MaybeUpdateNodeContextOutput() local
2407 ShapeHandle shape_handle; in UpdateQueue() local
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc501 shape_inference::ShapeHandle shape_handle = context->output(output_index); in BuildTensorShapeMapFromGraph() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc459 const ShapeHandle shape_handle, in CheckFormatConstraintsOnShape()