Searched defs:input_arrays (Results 1 – 13 of 13) sorted by relevance
37 void CopyTensorSegments(const std::vector<Array*>& input_arrays, in CopyTensorSegments()80 void ConcatenateTensorBuffers(const std::vector<Array*>& input_arrays, in ConcatenateTensorBuffers()111 const std::vector<Array*>& input_arrays, in SetMinMaxForConcatenedArray()172 std::vector<Array*> input_arrays; in Run() local
46 const std::vector<std::string>& input_arrays, in GraphdefToMlirImport()106 const std::vector<std::string>& input_arrays, in GraphdefToMlirTranslateFunction()127 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToMlirTranslateFunction()244 const std::vector<std::string>& input_arrays, in GraphdefToSplattedMlirTranslateFunction()296 absl::string_view input_arrays, absl::string_view input_dtypes, in GraphdefToSplattedMlirTranslateFunction()
26 opt<std::string> input_arrays( variable
155 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
2241 input_arrays=None, argument2618 input_arrays, argument2712 input_arrays=None, argument2764 input_arrays=None, argument2828 input_arrays, argument2840 input_arrays=None, argument2855 input_arrays=None, argument
80 input_arrays=None, argument
221 input_arrays, argument
76 absl::string_view debug_info_file, absl::string_view input_arrays, in LoadFromGraphdefOrMlirSource()
144 repeated InputArray input_arrays = 1; field
129 Arg<std::string> input_arrays; member
144 InputArray input_arrays[3]; in TEST() local
111 absl::string_view input_arrays, absl::string_view input_dtypes, in LoadFromGraphdefOrMlirSource()
3900 absl::Span<HloInstruction* const> input_arrays = reduce_window->inputs(); in HandleReduceWindow() local