Searched refs:flattened_inputs (Results 1 – 4 of 4) sorted by relevance
2356 std::vector<tensorflow::Safe_PyObjectPtr>* flattened_inputs, argument2391 if (flattened_inputs != nullptr) {2392 flattened_inputs->emplace_back(std::move(py_eager_tensor));2443 const std::vector<tensorflow::Safe_PyObjectPtr>* const flattened_inputs, argument2448 tensorflow::Safe_PyObjectPtr inputs(PyTuple_New(flattened_inputs->size()));2449 for (int i = 0; i < flattened_inputs->size(); i++) {2450 PyObject* input = (*flattened_inputs)[i].get();2636 std::unique_ptr<std::vector<tensorflow::Safe_PyObjectPtr>> flattened_inputs = local2642 flattened_inputs.reset(new std::vector<tensorflow::Safe_PyObjectPtr>);2679 flattened_inputs.get(), op, status)) {[all …]
758 flattened_inputs = (761 per_host_sharded_inputs.append(flattened_inputs)904 flattened_inputs = (907 control_deps.extend(flattened_inputs)908 per_host_sharded_inputs.append(flattened_inputs)985 flattened_inputs = None # Cache result from input_fn.996 if flattened_inputs is None:1002 flattened_inputs = (1008 array_ops.split(x, num_replicas) for x in flattened_inputs1016 broadcasted_inputs.append(flattened_inputs)[all …]
837 flattened_inputs = per_host_sharded_inputs[replica_index]838 inputs_part_dims_flat = nest.flatten_up_to(flattened_inputs,
279 flattened_inputs = array_ops.reshape(inputs, [-1])290 array_ops.gather(flattened_inputs, flattened_tag_indices),