Home
last modified time | relevance | path

Searched defs:outer_dims (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dshape_partition.cc25 std::vector<int64> outer_dims; in Run() local
Dir_emitter.cc1993 std::vector<int64> outer_dims; in HandleSlice() local
2411 std::vector<int64> outer_dims(std::next(concat_dim_layout_itr), in EmitFastConcatenate() local
/external/tensorflow/tensorflow/core/kernels/
Dscatter_nd_op.cc100 const int64 outer_dims = indices.shape().dims() - 1; in Compute() local
171 const int64 outer_dims = indices.shape().dims() - 1; in Compute() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2885 const int64 outer_dims = c->Rank(indices_shape) - 1; in ScatterNdShapeHelper() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc1755 const int64 outer_dims = std::max(rhs_outer_dims, lhs_outer_dims); in HandleDot() local