Searched defs:flat_outputs (Results 1 – 7 of 7) sorted by relevance
734 def record(self, flat_outputs, inference_args, input_tangents): argument1191 def record(self, flat_outputs, inference_args, input_tangents): argument1382 def record(self, flat_outputs): argument
254 std::vector<std::pair<const Value*, const TensorDesc>> flat_outputs; in compileKernel() local
174 uint64_t flat_outputs; member
426 bool flat_outputs[MAX_SLOTS] = { 0 }; in nir_lower_io_to_vector_impl() local
398 std::vector<tensorflow::Tensor> flat_outputs; in Run() local
828 std::vector<tensorflow::Tensor> flat_outputs; in RunMultipleSignatures() local
2500 def _set_mask_keras_history_checked(self, flat_outputs): argument