| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/src/ |
| D | OperationsUtils.cpp | 41 bool SetShape(const Shape& in, Shape* out) { in SetShape() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | weights.cc | 49 Status TRT_ShapedWeights::SetShape(DimsAdapter dims) { in SetShape() function in tensorflow::tensorrt::convert::TRT_ShapedWeights
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | remote_tensor_handle_data.cc | 177 Status RemoteTensorHandleData::SetShape(const TensorShape& shape) { in SetShape() function in tensorflow::RemoteTensorHandleData
|
| /aosp_15_r20/external/armnn/include/armnn/ |
| H A D | Tensor.hpp | 193 void SetShape(const TensorShape& newShape) { m_Shape = newShape; } in SetShape() function in armnn::TensorInfo
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | scatter_nd_test.cc | 55 void SetShape(std::initializer_list<T> data) { in SetShape() function in tflite::__anon60c175e50111::ScatterNdOpModel
|
| H A D | broadcast_to_test.cc | 61 void SetShape(std::initializer_list<ShapeType> data) { in SetShape() function in tflite::__anonfd2984b30111::BroadcastToOpModel
|
| H A D | kernel_util_test.cc | 74 void SetShape(TfLiteTensor* tensor, std::initializer_list<int> dims) { in SetShape() function in tflite::__anonc351f59c0111::KernelUtilTest
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | test_util.cc | 51 void FlexModelTest::SetShape(int tensor_index, const std::vector<int>& values) { in SetShape() function in tflite::flex::testing::FlexModelTest
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | constant_map.cpp | 54 void ConstantValueMap::SetShape( in SetShape() function in torch::jit::ConstantValueMap
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | tf_tensor.cc | 238 void TensorInterface::SetShape(const int64_t* dims, int num_dims) { in SetShape() function in tensorflow::TensorInterface
|
| H A D | c_api_test.cc | 365 TEST(CAPI, SetShape) { in TEST() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | shape_refiner.cc | 273 Status ShapeRefiner::SetShape(const Node* node, int output_port, in SetShape() function in tensorflow::ShapeRefiner
|
| H A D | shape_refiner_test.cc | 171 TEST_F(ShapeRefinerTest, SetShape) { in TEST_F() argument
|