Searched refs:ATTR_OUTPUT_SHAPES (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.h | 50 static constexpr const char* const ATTR_OUTPUT_SHAPES = variable
|
| D | remote_fused_graph_execute_utils_test.cc | 390 *node_def, RemoteFusedGraphExecuteUtils::ATTR_OUTPUT_SHAPES, &shapes)); in TEST()
409 *node_def, RemoteFusedGraphExecuteUtils::ATTR_OUTPUT_SHAPES, &shapes)); in TEST()
|
| D | remote_fused_graph_execute_utils.cc | 149 RemoteFusedGraphExecuteUtils::ATTR_OUTPUT_SHAPES; member in tensorflow::RemoteFusedGraphExecuteUtils
370 AddNodeAttr(ATTR_OUTPUT_SHAPES, shapes, node_def); in AddOutputTensorShapeType()
411 status = GetNodeAttr(attrs, ATTR_OUTPUT_SHAPES, shapes); in GetOutputTensorShapeType()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 349 .Attr(RemoteFusedGraphExecuteUtils::ATTR_OUTPUT_SHAPES, shapes); in TransformGraphToAddAggregatedInputNode()
390 created_node->AddAttr(RemoteFusedGraphExecuteUtils::ATTR_OUTPUT_SHAPES, in TransformGraphToAddAggregatedInputNode()
|