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.hpp44 const unsigned int shape0[NumDims], in ElementwiseTestHelper()
48 const unsigned int shape1[NumDims], in ElementwiseTestHelper()
52 const unsigned int outShape[NumDims], in ElementwiseTestHelper()
115 const unsigned int shape0[NumDims], in ElementwiseTestHelper()
119 const unsigned int shape1[NumDims], in ElementwiseTestHelper()
123 const unsigned int outShape[NumDims], in ElementwiseTestHelper()
156 const unsigned int shape0[NumDims], in ElementwiseTestHelper()
158 const unsigned int shape1[NumDims], in ElementwiseTestHelper()
160 const unsigned int outShape[NumDims], in ElementwiseTestHelper()
191 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()
95 const unsigned int shape[NumDims], in ElementwiseUnaryTestHelper()
97 const unsigned int outShape[NumDims], in ElementwiseUnaryTestHelper()
/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h123 static const int NumDims = internal::traits<OutputBackward>::NumDimensions; variable
426 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/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h115 const int NumDims = internal::traits<Input>::NumDimensions; variable
/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::__anon325885840111::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.cc133 Status RemoteTensorHandleData::NumDims(int* num_dims) const { in NumDims() function in tensorflow::RemoteTensorHandleData
/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.cc215 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::__anon9dfae7e90111::FakeTensorHandle
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.cc305 int TensorInterface::NumDims() const { return tensor_.get().shape().GetRank(); } in NumDims() function in tfrt::tf::TensorInterface
406 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.cc502 Status NumDims(int* num_dims) const override { in NumDims() function in tensorflow::__anon9d665ed60211::CAPICustomDeviceTensorHandle