Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A 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 …]
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_backward_spatial_convolutions.h118 static const int NumDims = internal::traits<OutputBackward>::NumDimensions; variable
406 static const int NumDims = internal::traits<Input>::NumDimensions; variable
H A Deigen_backward_cuboid_convolutions.h112 static const int NumDims = internal::traits<OutputBackward>::NumDimensions; variable
412 static const int NumDims = internal::traits<Input>::NumDimensions; variable
H A 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
H A 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
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/include/clang/AST/
DExprOpenMP.h31 unsigned NumDims = 0; /// Number of dimensions in the shaping expression. variable
38 explicit OMPArrayShapingExpr(EmptyShell Shell, unsigned NumDims) in OMPArrayShapingExpr()
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/include/clang/AST/
DExprOpenMP.h155 unsigned NumDims = 0; /// Number of dimensions in the shaping expression. variable
162 explicit OMPArrayShapingExpr(EmptyShell Shell, unsigned NumDims) in OMPArrayShapingExpr()
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/include/clang/AST/
DExprOpenMP.h155 unsigned NumDims = 0; /// Number of dimensions in the shaping expression. variable
162 explicit OMPArrayShapingExpr(EmptyShell Shell, unsigned NumDims) in OMPArrayShapingExpr()
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/include/clang/AST/
DExprOpenMP.h155 unsigned NumDims = 0; /// Number of dimensions in the shaping expression. variable
162 explicit OMPArrayShapingExpr(EmptyShell Shell, unsigned NumDims) in OMPArrayShapingExpr()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dtensor_handle_data.cc50 Status LocalTensorHandleData::NumDims(int* num_dims) const { in NumDims() function in tensorflow::LocalTensorHandleData
H A Dcustom_device_test.cc75 Status NumDims(int* num_dims) const override { in NumDims() function in tensorflow::eager::__anonb9701f0e0111::TestCustomDeviceTensorHandle
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Dremote_tensor_handle_data.cc134 Status RemoteTensorHandleData::NumDims(int* num_dims) const { in NumDims() function in tensorflow::RemoteTensorHandleData
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dutils.h283 int32_t NumDims() const { return num_dims_; } in NumDims() function
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorMap.h262 const std::size_t NumDims = sizeof...(otherIndices) + 2; in operator() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dtf_tensor.cc220 int TensorInterface::NumDims() const { return tensor_.dims(); } in NumDims() function in tensorflow::TensorInterface
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
H A Dop_handler_selector_test.cc122 Status NumDims(int* num_dims) const override { in NumDims() function in tfrt::tf::__anon6788e1330111::FakeTensorHandle
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_block_eval.cpp593 static void VerifyBlockAssignment(Tensor<T, NumDims, Layout>& tensor, in VerifyBlockAssignment() argument
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api.cc504 Status NumDims(int* num_dims) const override { in NumDims() function in tensorflow::__anon816881a00211::CAPICustomDeviceTensorHandle