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
|