| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node.cc | 21 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs) { in Init() argument
27 const int n_inputs = op_inputs.size(); in Init()
31 TensorHandle* const* op_inputs_array = &op_inputs[0]; in Init()
72 [ctx, &op_inputs](const int i, in Init()
75 op_inputs[i], handle, op_inputs[i]->device(), in Init()
76 op_inputs[i]->device()->name()); in Init()
|
| D | execute.h | 51 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs,
|
| D | execute_node.h | 52 const gtl::InlinedVector<TensorHandle*, 4>& op_inputs);
|
| D | execute.cc | 945 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs, in EagerKernelExecute() argument
954 ExecuteNodeArgs inputs(op_inputs.size()); in EagerKernelExecute()
955 TF_RETURN_IF_ERROR(inputs.Init(ctx, op_inputs)); in EagerKernelExecute()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | node_info_delegate_test.cc | 108 auto op_inputs = operators->Get(i)->inputs(); in TEST() local
110 info.inputs = std::vector<int>(op_inputs->begin(), op_inputs->end()); in TEST()
|
| D | calibrator.cc | 344 auto op_inputs = op->inputs(); in BuildLoggingInterpreter() local
346 op_info.inputs = std::vector<int>(op_inputs->begin(), op_inputs->end()); in BuildLoggingInterpreter()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | op_selector.py | 230 op_inputs = op.inputs
232 return op_inputs if is_differentiable(op) else []
234 return op_inputs
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 117 absl::Span<const RefPtr<XRTTupleAllocation>> op_inputs) { in GetChainedOpInputs() argument
124 input_buffers.input_tuples.emplace_back(op_inputs[i]); in GetChainedOpInputs()
372 absl::Span<const RefPtr<XRTTupleAllocation>> op_inputs) in DoWork()
375 GetChainedOpInputs(op, op_inputs)); in DoWork()
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2_test.py | 915 op_inputs = list(op.inputs) + op.control_inputs
916 names = [i.name for i in op_inputs]
917 op_inputs = [getattr(i, 'op', i) for i in op_inputs]
918 return op_inputs, names
936 op_inputs = get_inputs(op)[0]
937 if not op_inputs:
940 inputs[op] = set(op_inputs)
941 for i in op_inputs:
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | interpreter.py | 285 op_inputs = self._interpreter.NodeInputs(op_index)
291 'inputs': op_inputs,
|
| /external/XNNPACK/bench/ |
| D | prelu.cc | 152 const int32_t op_inputs[2] = { 0, 1 }; in tflite_prelu_f32() local
157 builder.CreateVector<int32_t>(op_inputs, 2), in tflite_prelu_f32()
|
| D | softmax.cc | 197 const int32_t op_inputs[1] = { 0 }; in tflite_softmax_f32() local
202 builder.CreateVector<int32_t>(op_inputs, 1), in tflite_softmax_f32()
|
| D | sigmoid.cc | 195 const int32_t op_inputs[1] = { 0 }; in tflite_sigmoid_f32() local
200 builder.CreateVector<int32_t>(op_inputs, 1), in tflite_sigmoid_f32()
|
| D | deconvolution.cc | 369 const int32_t op_inputs[3] = { 0, 1, 2 }; in tflite_deconvolution_f32() local
374 builder.CreateVector<int32_t>(op_inputs, 3), in tflite_deconvolution_f32()
|
| D | convolution.cc | 399 const int32_t op_inputs[3] = { 0, 1, 2 }; in tflite_convolution_f32() local
404 builder.CreateVector<int32_t>(op_inputs, 3), in tflite_convolution_f32()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 226 op_inputs = list(op.inputs)
229 for i, op_input in enumerate(op_inputs):
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 3112 PyObject* op_inputs; local
3118 op_inputs = Py_None;
3121 op_inputs =
3125 op_inputs = inputs;
3154 [op_name, attrs, num_inputs, op_inputs, op_outputs]() {
3158 Py_INCREF(op_inputs);
3161 [op_name, attrs, num_inputs, op_inputs, op_outputs](
3181 "OOOOOOO", op_name, attrs, num_inputs, op_inputs, op_outputs,
3193 [op_name, attrs, num_inputs, op_inputs,
3198 Py_DECREF(op_inputs);
[all …]
|