Home
last modified time | relevance | path

Searched defs:SetShape (Results 1 – 10 of 10) sorted by relevance

/external/armnn/include/armnn/
DTensor.hpp189 void SetShape(const TensorShape& newShape) { m_Shape = newShape; } in SetShape() function in armnn::TensorInfo
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_tensor_handle_data.cc176 Status RemoteTensorHandleData::SetShape(const TensorShape& shape) { in SetShape() function in tensorflow::RemoteTensorHandleData
/external/tensorflow/tensorflow/lite/kernels/
Dscatter_nd_test.cc55 void SetShape(std::initializer_list<T> data) { in SetShape() function in tflite::__anon0007b05b0111::ScatterNdOpModel
Dbroadcast_to_test.cc61 void SetShape(std::initializer_list<ShapeType> data) { in SetShape() function in tflite::__anon86e834a90111::BroadcastToOpModel
Dkernel_util_test.cc74 void SetShape(TfLiteTensor* tensor, std::initializer_list<int> dims) { in SetShape() function in tflite::__anon4b5f7ed20111::KernelUtilTest
/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.cc51 void FlexModelTest::SetShape(int tensor_index, const std::vector<int>& values) { in SetShape() function in tflite::flex::testing::FlexModelTest
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc274 Status ShapeRefiner::SetShape(const Node* node, int output_port, in SetShape() function in tensorflow::ShapeRefiner
Dshape_refiner_test.cc169 TEST_F(ShapeRefinerTest, SetShape) { in TEST_F() argument
/external/tensorflow/tensorflow/c/
Dc_api_test.cc365 TEST(CAPI, SetShape) { in TEST() argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc811 Status TRT_ShapedWeights::SetShape(nvinfer1::Dims dims) { in SetShape() function in tensorflow::tensorrt::convert::TRT_ShapedWeights