Searched refs:flattened_inputs (Results 1 – 2 of 2) sorted by relevance
3417 std::vector<tensorflow::Safe_PyObjectPtr>* flattened_inputs, argument3451 if (flattened_inputs != nullptr) {3452 flattened_inputs->emplace_back(std::move(py_eager_tensor));3495 const std::vector<tensorflow::Safe_PyObjectPtr>& flattened_inputs, argument3500 tensorflow::Safe_PyObjectPtr inputs(PyTuple_New(flattened_inputs.size()));3501 for (int i = 0; i < flattened_inputs.size(); i++) {3502 PyObject* input = flattened_inputs[i].get();3710 std::unique_ptr<std::vector<tensorflow::Safe_PyObjectPtr>> flattened_inputs = local3716 flattened_inputs.reset(new std::vector<tensorflow::Safe_PyObjectPtr>);3752 flattened_attrs.get(), flattened_inputs.get(), op,[all …]
862 flattened_inputs = sharded_inputs[replica_index]863 inputs_part_dims_flat = nest.flatten_up_to(flattened_inputs,