Home
last modified time | relevance | path

Searched defs:NumDims (Results 1 – 17 of 17) sorted by relevance

/external/armnn/src/backends/backendsCommon/test/layerTests/
DElementwiseTestImpl.hpp43 const unsigned int shape0[NumDims], in ElementwiseTestHelper()
47 const unsigned int shape1[NumDims], in ElementwiseTestHelper()
51 const unsigned int outShape[NumDims], in ElementwiseTestHelper()
114 const unsigned int shape0[NumDims], in ElementwiseTestHelper()
118 const unsigned int shape1[NumDims], in ElementwiseTestHelper()
122 const unsigned int outShape[NumDims], in ElementwiseTestHelper()
154 const unsigned int shape0[NumDims], in ElementwiseTestHelper()
156 const unsigned int shape1[NumDims], in ElementwiseTestHelper()
158 const unsigned int outShape[NumDims], in ElementwiseTestHelper()
188 const unsigned int shape0[NumDims], in ElementwiseTestHelper()
[all …]
DElementwiseUnaryTestImpl.hpp39 const unsigned int shape[NumDims], in ElementwiseUnaryTestHelper()
43 const unsigned int outShape[NumDims], in ElementwiseUnaryTestHelper()
97 const unsigned int shape[NumDims], in ElementwiseUnaryTestHelper()
99 const unsigned int outShape[NumDims], in ElementwiseUnaryTestHelper()
/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h118 static const int NumDims = internal::traits<OutputBackward>::NumDimensions; variable
406 static const int NumDims = internal::traits<Input>::NumDimensions; variable
Deigen_backward_cuboid_convolutions.h112 static const int NumDims = internal::traits<OutputBackward>::NumDimensions; variable
412 static const int NumDims = internal::traits<Input>::NumDimensions; variable
Deigen_spatial_convolutions-inl.h116 const size_t NumDims = tensor.impl().dimensions().size(); in TensorContractionInputMapper() local
144 const int NumDims = tensor.impl().impl().dimensions().size(); in TensorContractionInputMapper() local
Deigen_cuboid_convolution.h113 const int NumDims = tensor.impl().dimensions().size(); in TensorContractionInputMapper() local
159 const int NumDims = tensor.impl().impl().dimensions().size(); in TensorContractionInputMapper() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle_data.cc50 Status LocalTensorHandleData::NumDims(int* num_dims) const { in NumDims() function in tensorflow::LocalTensorHandleData
Dcustom_device_test.cc75 Status NumDims(int* num_dims) const override { in NumDims() function in tensorflow::eager::__anona3a8ac9d0111::TestCustomDeviceTensorHandle
Dtensor_handle.cc86 Status TensorHandle::PackedTensorHandleData::NumDims(int* num_dims) const { in NumDims() function in tensorflow::TensorHandle::PackedTensorHandleData
627 Status TensorHandle::NumDims(int* num_dims) const { in NumDims() function in tensorflow::TensorHandle
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_tensor_handle_data.cc134 Status RemoteTensorHandleData::NumDims(int* num_dims) const { in NumDims() function in tensorflow::RemoteTensorHandleData
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h283 int32_t NumDims() const { return num_dims_; } in NumDims() function
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorMap.h262 const std::size_t NumDims = sizeof...(otherIndices) + 2; in operator() local
/external/tensorflow/tensorflow/c/
Dtf_tensor.cc220 int TensorInterface::NumDims() const { return tensor_.dims(); } in NumDims() function in tensorflow::TensorInterface
/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
Dop_handler_selector_test.cc122 Status NumDims(int* num_dims) const override { in NumDims() function in tfrt::tf::__anond52f36620111::FakeTensorHandle
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.cc306 int TensorInterface::NumDims() const { return tensor_.get().shape().GetRank(); } in NumDims() function in tfrt::tf::TensorInterface
447 tensorflow::Status TensorHandleInterface::NumDims(int* num_dims) const { in NumDims() function in tfrt::tf::TensorHandleInterface
/external/eigen/unsupported/test/
Dcxx11_tensor_block_eval.cpp593 static void VerifyBlockAssignment(Tensor<T, NumDims, Layout>& tensor, in VerifyBlockAssignment() argument
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc504 Status NumDims(int* num_dims) const override { in NumDims() function in tensorflow::__anon6338c50f0211::CAPICustomDeviceTensorHandle