Home
last modified time | relevance | path

Searched refs:op_inputs (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dnode_info_delegate_test.cc108 auto op_inputs = operators->Get(i)->inputs(); in TEST() local
110 info.inputs = std::vector<int>(op_inputs->begin(), op_inputs->end()); in TEST()
Dcalibrator.cc296 auto op_inputs = op->inputs(); in BuildLoggingInterpreter() local
298 op_info.inputs = std::vector<int>(op_inputs->begin(), op_inputs->end()); in BuildLoggingInterpreter()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.h47 const gtl::InlinedVector<TensorHandle*, 4>& op_inputs,
Dexecute.cc913 const gtl::InlinedVector<TensorHandle*, 4>& op_inputs, in EagerKernelExecute() argument
926 for (int i = 0; i < op_inputs.size(); ++i) { in EagerKernelExecute()
927 if (!op_inputs[i]->RefCountIsOne()) { in EagerKernelExecute()
929 TF_RETURN_IF_ERROR(op_inputs[i]->Tensor(&input_tensor)); in EagerKernelExecute()
934 gtl::InlinedVector<TensorValue, 4> input_vector(op_inputs.size()); in EagerKernelExecute()
935 for (int i = 0; i < op_inputs.size(); ++i) { in EagerKernelExecute()
936 TF_RETURN_IF_ERROR(op_inputs[i]->TensorValue(&input_vector[i])); in EagerKernelExecute()
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_utils.py277 op_inputs = list(op.inputs)
280 for i, op_input in enumerate(op_inputs):
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc2108 PyObject* op_inputs; local
2114 op_inputs = Py_None;
2117 op_inputs =
2121 op_inputs = inputs;
2128 [op_name, attrs, num_inputs, op_inputs, op_outputs]() {
2132 Py_INCREF(op_inputs);
2135 new PyBackwardFunction([op_name, attrs, num_inputs, op_inputs,
2141 Py_BuildValue("OOOOOO", op_name, attrs, num_inputs, op_inputs,
2153 [op_name, attrs, num_inputs, op_inputs,
2158 Py_DECREF(op_inputs);
[all …]
/external/tensorflow/tensorflow/contrib/recurrent/python/ops/
Drecurrent.py389 op_inputs = [x for x in op.inputs]
394 (theta, state0, inputs, max_input_length, _) = _Pack(op_inputs, op_struct)