Searched defs:inner_dim (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cross_op.cc | 60 auto inner_dim = in0.dim_size(in0.dims() - 1); in Compute() local
|
| D | redux_functor.h | 51 Eigen::Index inner_dim = 1, outer_dim = 1; in operator() local
220 Eigen::Index inner_dim = 1, middle_dim = 1, outer_dim = 1; in operator() local
|
| D | count_ops.cc | 50 int inner_dim = is_1d ? 1 : 2; in OutputSparse() local
|
| D | training_ops.cc | 1464 int64_t inner_dim = 1; in Compute() local
1924 int64_t inner_dim = 1; in Compute() local
2058 int64_t inner_dim = 1; in Compute() local
2207 int64_t inner_dim = 1; in Compute() local
2464 int64_t inner_dim = 1; in Compute() local
2890 int64_t inner_dim = 1; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cross_op.cc | 39 auto inner_dim = in0_shape.dim_size(in0_shape.dims() - 1); in Compile() local
|
| /external/libaom/aom_dsp/ |
| D | mathutils.h | 96 const int m1_rows, const int inner_dim, in multiply_mat()
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | segment_reduction_ops_test.py | 1098 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_block_eval.cpp | 328 const int inner_dim = Layout == static_cast<int>(ColMajor) ? 0 : NumDims - 1; in test_eval_tensor_padding() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2667 [&](int64_t product, int64_t inner_dim) { in EmitFastConcatenate()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1038 int64_t inner_dim = RandomDim(); in ChooseBatchMatMulArguments() local
1102 int64_t inner_dim = RandomDim(); in ChooseEinsumArguments() local
|