Searched defs:in_dims (Results 1 – 7 of 7) sorted by relevance
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
D | TensorBase.h | 663 DimensionList<Index, NumDimensions> in_dims; in sum() local 675 DimensionList<Index, NumDimensions> in_dims; in mean() local 687 DimensionList<Index, NumDimensions> in_dims; in prod() local 700 DimensionList<Index, NumDimensions> in_dims; in maximum() local 713 DimensionList<Index, NumDimensions> in_dims; in minimum() local 726 DimensionList<Index, NumDimensions> in_dims; in all() local 739 DimensionList<Index, NumDimensions> in_dims; in any() local 748 array<Index, NumDimensions> in_dims; in argmax() local 761 array<Index, NumDimensions> in_dims; in argmin() local 774 array<Index, 1> in_dims; in argmax() local [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | runtime_fft_impl.h | 76 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; in EigenFftR2C() local 115 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; in EigenFftC2R() local
|
/external/ComputeLibrary/arm_compute/core/ |
D | Helpers.h | 211 inline Size2D compute_winograd_convolution_tiles(const Size2D &in_dims, const Size2D &kernel_size, … in compute_winograd_convolution_tiles()
|
/external/tensorflow/tensorflow/core/kernels/mkl/ |
D | mkl_transpose_op.cc | 72 memory::dims in_dims = TFShapeToMklDnnDims(in_tensor.shape()); in MKLTransposeND() local
|
/external/tensorflow/tensorflow/core/util/ |
D | mkl_util.h | 1123 inline memory::dims MklDnnDimsInNCHW(const memory::dims& in_dims, in MklDnnDimsInNCHW() 1140 inline memory::dims MklDnnDimsInNCDHW(const memory::dims& in_dims, in MklDnnDimsInNCDHW()
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 2010 std::vector<int64> in_dims = RandomDims(); in TEST_F() local 2691 std::vector<int64> in_dims = RandomDims(4, 4); in TEST_F() local 2706 std::vector<int64> in_dims = RandomDims(4, 4); in TEST_F() local
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | algebraic_simplifier_test.cc | 4120 std::vector<int64> in_dims; in TEST_F() local
|