Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node.cc65 EagerContext* ctx, const gtl::InlinedVector<TensorHandle*, 4>& op_inputs, in Init() argument
72 const int n_inputs = op_inputs.size(); in Init()
74 TensorHandle* const* op_inputs_flat = &op_inputs[0]; in Init()
101 [ctx, &op_inputs, is_function]( in Init()
104 TensorHandle* h = op_inputs[index.index]; in Init()
105 if (op_inputs[index.index]->Type() == TensorHandle::PACKED) { in Init()
107 op_inputs[index.index]->ExtractPackedHandle(index.sub_index, &h)); in Init()
Dexecute.h51 EagerContext* ctx, const absl::InlinedVector<TensorHandle*, 4>& op_inputs,
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreshape_tester.cc136 std::vector<int32_t> op_inputs({0}); in CreateTfLiteModel() local
138 op_inputs.push_back(1); in CreateTfLiteModel()
155 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in CreateTfLiteModel()
159 const std::array<int32_t, 1> subgraph_inputs{{op_inputs.front()}}; in CreateTfLiteModel()
Ddepth_to_space_tester.cc139 const std::array<int32_t, 1> op_inputs{{0}}; in CreateTfLiteModel() local
143 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in CreateTfLiteModel()
148 const std::array<int32_t, 1> subgraph_inputs{{op_inputs.front()}}; in CreateTfLiteModel()
Dleaky_relu_tester.cc121 const std::array<int32_t, 1> op_inputs{{0}}; in CreateTfLiteModel() local
125 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in CreateTfLiteModel()
Dsoftmax_tester.cc123 const std::array<int32_t, 1> op_inputs{{0}}; in CreateTfLiteModel() local
127 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in CreateTfLiteModel()
Dunary_elementwise_tester.cc149 const std::array<int32_t, 1> op_inputs{{0}}; in CreateTfLiteModel() local
153 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in CreateTfLiteModel()
Dreduce_tester.cc138 const std::array<int32_t, 2> op_inputs{{0, 1}}; in CreateTfLiteModel() local
142 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in CreateTfLiteModel()
Dresize_bilinear_tester.cc155 const std::array<int32_t, 2> op_inputs{{0, 1}}; in CreateTfLiteModel() local
159 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in CreateTfLiteModel()
Dpad_tester.cc155 const std::array<int32_t, 2> op_inputs{{0, 1}}; in CreateTfLiteModel() local
159 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in CreateTfLiteModel()
Dpool_2d_tester.cc167 const std::array<int32_t, 1> op_inputs{{0}}; in CreateTfLiteModel() local
172 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in CreateTfLiteModel()
Dprelu_tester.cc199 const std::array<int32_t, 2> op_inputs{ in CreateTfLiteModel() local
206 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in CreateTfLiteModel()
Dfully_connected_tester.cc247 const std::array<int32_t, 3> op_inputs{ in CreateTfLiteModel() local
255 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in CreateTfLiteModel()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/
Dadd.cc57 const int32_t op_inputs[2] = {0, 1}; in Build() local
62 builder.CreateVector<int32_t>(op_inputs, 2), in Build()
/external/XNNPACK/bench/
Dbankers-rounding.cc133 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_bankers_rounding_f32() local
138 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in tflite_bankers_rounding_f32()
Delu.cc133 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_elu_f32() local
138 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in tflite_elu_f32()
Dceiling.cc133 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_ceiling_f32() local
138 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in tflite_ceiling_f32()
Dsquare-root.cc133 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_square_root_f32() local
138 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in tflite_square_root_f32()
Dfloor.cc133 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_floor_f32() local
138 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in tflite_floor_f32()
Dsigmoid.cc208 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_sigmoid_f32() local
213 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in tflite_sigmoid_f32()
Dhardswish.cc135 const std::array<int32_t, 1> op_inputs{{ 0 }}; in tflite_hardswish_f32() local
140 builder.CreateVector<int32_t>(op_inputs.data(), op_inputs.size()), in tflite_hardswish_f32()
/external/tensorflow/tensorflow/python/ops/
Dop_selector.py230 op_inputs = op.inputs
232 return op_inputs if is_differentiable(op) else []
234 return op_inputs
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc103 absl::Span<const RefPtr<XRTTupleAllocation>> op_inputs) { in GetChainedOpInputTuples() argument
112 input_tuples.emplace_back(op_inputs[i]); in GetChainedOpInputTuples()
116 op_inputs[i].get(), {input.output_index() - 1}, &sub_tuple, in GetChainedOpInputTuples()
559 absl::Span<const RefPtr<XRTTupleAllocation>> op_inputs) in DoWork()
566 GetChainedOpInputTuples(op, op_inputs)); in DoWork()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator.cc370 auto op_inputs = op->inputs(); in BuildLoggingInterpreter() local
372 op_info.inputs = std::vector<int>(op_inputs->begin(), op_inputs->end()); in BuildLoggingInterpreter()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc274 absl::InlinedVector<tensorflow::TensorHandle*, 4>* op_inputs; in BuildEagerInputs() local
275 TF_RETURN_IF_ERROR(op_->MutableTensorHandleInputs(&op_inputs)); in BuildEagerInputs()
290 op_inputs->push_back(handle); in BuildEagerInputs()
297 op_inputs->push_back(handle); in BuildEagerInputs()

12