Searched defs:in_dims (Results 1 – 8 of 8) 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 | 68 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; in EigenFftR2C() local
107 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> in_dims; in EigenFftC2R() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_transpose_op.cc | 131 memory::dims in_dims = TFShapeToMklDnnDims(in_tensor.shape()); in MKLTransposeND() local
|
| D | mkl_conv_grad_bias_ops.cc | 148 int in_dims; member
|
| D | mkl_conv_ops.cc | 686 int in_dims; member
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2009 std::vector<int64> in_dims = RandomDims(); in TEST_F() local
2690 std::vector<int64> in_dims = RandomDims(4, 4); in TEST_F() local
2705 std::vector<int64> in_dims = RandomDims(4, 4); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1549 inline memory::dims MklDnnDimsInNCHW(const memory::dims& in_dims, in MklDnnDimsInNCHW()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 3323 std::vector<int64> in_dims; in TEST_F() local
|