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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
resolve_constant_concatenation.cc 31 // concatenation dimension. From each array in input_arrays, it copies chunk
35 void CopyTensorSegments(const std::vector<Array*>& input_arrays,
39 for (Array* input_array : input_arrays) {
51 CHECK(!input_arrays.empty());
54 input_arrays[0]->GetBuffer<A>().data.size() / array_copy_size[0];
58 src_ptr.reserve(input_arrays.size());
59 for (Array* input_array : input_arrays) {
63 // Copy the data from input_arrays to concatenated_array_buffer.
66 for (int i = 0; i < input_arrays.size(); i++) {
78 void ConcatenateTensorBuffers(const std::vector<Array*>& input_arrays,
159 std::vector<Array*> input_arrays; local
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/
args.h 189 Arg<string> input_arrays; member in struct:toco::ParsedModelFlags

Completed in 220 milliseconds