Searched refs:flattened_inputs (Results 1 – 2 of 2) sorted by relevance
3500 std::vector<tensorflow::Safe_PyObjectPtr>* flattened_inputs, argument3534 if (flattened_inputs != nullptr) {3535 flattened_inputs->emplace_back(std::move(py_eager_tensor));3578 const std::vector<tensorflow::Safe_PyObjectPtr>& flattened_inputs, argument3583 tensorflow::Safe_PyObjectPtr inputs(PyTuple_New(flattened_inputs.size()));3584 for (int i = 0; i < flattened_inputs.size(); i++) {3585 PyObject* input = flattened_inputs[i].get();3793 std::unique_ptr<std::vector<tensorflow::Safe_PyObjectPtr>> flattened_inputs = local3799 flattened_inputs.reset(new std::vector<tensorflow::Safe_PyObjectPtr>);3835 flattened_attrs.get(), flattened_inputs.get(), op,[all …]
863 flattened_inputs = sharded_inputs[replica_index]864 inputs_part_dims_flat = nest.flatten_up_to(flattened_inputs,