Searched defs:in_dims (Results 1 – 6 of 6) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBase.h | 532 DimensionList<Index, NumDimensions> in_dims; in sum() local
544 DimensionList<Index, NumDimensions> in_dims; in mean() local
556 DimensionList<Index, NumDimensions> in_dims; in prod() local
568 DimensionList<Index, NumDimensions> in_dims; in maximum() local
580 DimensionList<Index, NumDimensions> in_dims; in minimum() local
593 DimensionList<Index, NumDimensions> in_dims; in all() local
606 DimensionList<Index, NumDimensions> in_dims; in any() local
615 array<Index, NumDimensions> in_dims; in argmax() local
628 array<Index, NumDimensions> in_dims; in argmin() local
641 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/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_transpose_op.cc | 132 memory::dims in_dims = TFShapeToMklDnnDims(in_tensor.shape()); in MKLTransposeND() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1120 inline memory::dims MklDnnDimsInNCHW(const memory::dims& in_dims, in MklDnnDimsInNCHW()
1137 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 | 4087 std::vector<int64> in_dims; in TEST_F() local
|