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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_constant_folding_test.cc 123 int64 concat_size = 0; local
127 concat_size += csize;
133 dimensions[test_config.concat_dimension] = concat_size;
  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
propagate_fixed_sizes.cc 675 int concat_size = 0; local
686 concat_size += input_dims[op->axis];
688 // Write out the concat_size on the output array shape.
692 output_dims[op->axis] = concat_size;
    [all...]

Completed in 341 milliseconds