Home
last modified time | relevance | path

Searched refs:TensorShapeType (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.h90 using TensorShapeType = std::pair<DataType, TensorShape>; variable
93 TensorShapeType>>;
158 static const TensorShapeType* GetTensorShapeType(
161 static const TensorShapeType* GetTensorShapeType(
Dremote_fused_graph_execute_utils.cc378 std::priority_queue<std::tuple<int, const TensorShapeType*>> queue; in AddOutputTensorShapeTypeByTensorShapeMap()
388 const TensorShapeType* tst = std::get<1>(queue.top()); in AddOutputTensorShapeTypeByTensorShapeMap()
520 /* static */ const RemoteFusedGraphExecuteUtils::TensorShapeType*
531 /* static */ const RemoteFusedGraphExecuteUtils::TensorShapeType*
Dremote_fused_graph_execute_utils_test.cc317 const RemoteFusedGraphExecuteUtils::TensorShapeType* tst = in TEST()
363 const RemoteFusedGraphExecuteUtils::TensorShapeType* tst = in TEST()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc201 template <typename TensorShapeType>
202 inline nvinfer1::Dims TensorShapeToTrtDims(const TensorShapeType& shape, in TensorShapeToTrtDims()