HomeSort by relevance Sort by last modified time
    Searched defs:output_rank (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
instruction_fusion.cc 135 int64 output_rank = 0; local
138 [&output_rank](const Shape& subshape, const ShapeIndex& shape_index) {
140 output_rank = std::max(output_rank, ShapeUtil::TrueRank(subshape));
144 [output_rank](HloInstruction* operand) {
153 output_rank;
  /external/tensorflow/tensorflow/core/kernels/
reshape_util.cc 51 const int64 output_rank = target_shape_in.NumElements(); local
62 for (int d = 0; d < output_rank; ++d) {
115 gtl::InlinedVector<int64, 8> output_strides(output_rank);
116 output_strides[output_rank - 1] = 1;
117 for (int d = output_rank - 2; d >= 0; --d) {
124 TensorShape({nnz, output_rank}),
133 for (int j = 0; j < output_rank; ++j) {
141 TensorShape({output_rank}),
  /external/tensorflow/tensorflow/contrib/lite/kernels/
embedding_lookup_sparse.cc 154 const int output_rank = (lookup_rank - 1) + (embedding_rank - 1); local
161 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(output_rank);

Completed in 85 milliseconds