Home
last modified time | relevance | path

Searched defs:out_dim_sizes (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_reduce_op.cc86 std::vector<int64> out_dim_sizes; in SparseTensorReduceHelper() local
322 auto out_dim_sizes = reduction.reduced_shape.dim_sizes(); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc476 const LocalOp& operand, absl::Span<const int64> out_dim_sizes, in BroadcastInDim()