Home
last modified time | relevance | path

Searched defs:inner_dim (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcross_op.cc60 auto inner_dim = in0.dim_size(in0.dims() - 1); in Compute() local
Dtraining_ops.cc938 int64 inner_dim = 1; in Compute() local
1278 int64 inner_dim = 1; in Compute() local
1429 int64 inner_dim = 1; in Compute() local
1718 int64 inner_dim = 1; in Compute() local
2070 int64 inner_dim = 1; in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcross_op.cc38 auto inner_dim = in0_shape.dim_size(in0_shape.dims() - 1); in Compile() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dsegment_reduction_ops_test.py871 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
/external/tensorflow/tensorflow/core/ops/
Dmath_ops.cc1523 auto inner_dim = c->Dim(input, rank - 1); in __anon2aeb9a7a1202() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2386 [&](int64 product, int64 inner_dim) { in EmitFastConcatenate()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1116 int64 inner_dim = RandomDim(); in TEST_F() local