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 | 41 int inner_dim = is_1d ? 1 : 2; in OutputSparse() local
|
| D | training_ops.cc | 1424 int64 inner_dim = 1; in Compute() local
1892 int64 inner_dim = 1; in Compute() local
2026 int64 inner_dim = 1; in Compute() local
2175 int64 inner_dim = 1; in Compute() local
2432 int64 inner_dim = 1; in Compute() local
2858 int64 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/libaom/av1/encoder/ |
| D | mathutils.h | 94 const int m1_rows, const int inner_dim, in multiply_mat()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | segment_reduction_ops_test.py | 975 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops.cc | 1942 auto inner_dim = c->Dim(input, rank - 1); in __anon725c8f641702() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2506 [&](int64 product, int64 inner_dim) { in EmitFastConcatenate()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1205 int64 inner_dim = RandomDim(); in TEST_F() local
|