Home
last modified time | relevance | path

Searched refs:GetInput (Results 1 – 25 of 125) sorted by relevance

12345

/external/tensorflow/tensorflow/core/kernels/
Dparse_tensor_test.cc87 test::ExpectTensorEqual<Eigen::half>(parse_output, GetInput(0)); in TEST_F()
96 test::ExpectTensorEqual<float>(parse_output, GetInput(0)); in TEST_F()
105 test::ExpectTensorEqual<double>(parse_output, GetInput(0)); in TEST_F()
114 test::ExpectTensorEqual<int64>(parse_output, GetInput(0)); in TEST_F()
123 test::ExpectTensorEqual<int32>(parse_output, GetInput(0)); in TEST_F()
132 test::ExpectTensorEqual<int16>(parse_output, GetInput(0)); in TEST_F()
141 test::ExpectTensorEqual<int8>(parse_output, GetInput(0)); in TEST_F()
150 test::ExpectTensorEqual<uint16>(parse_output, GetInput(0)); in TEST_F()
159 test::ExpectTensorEqual<uint8>(parse_output, GetInput(0)); in TEST_F()
169 test::ExpectTensorEqual<complex64>(parse_output, GetInput(0)); in TEST_F()
[all …]
Dfake_quant_ops_test.cc380 auto input_flat = GetInput(0).flat<float>(); in TEST_F()
410 auto input_flat = GetInput(0).flat<float>(); in TEST_F()
441 auto input_flat = GetInput(0).flat<float>(); in TEST_F()
472 auto input_flat = GetInput(0).flat<float>(); in TEST_F()
670 auto in_flat = GetInput(0).flat<float>(); in TEST_F()
715 auto in_flat = GetInput(0).flat<float>(); in TEST_F()
759 auto in_flat = GetInput(0).flat<float>(); in TEST_F()
804 auto in_flat = GetInput(0).flat<float>(); in TEST_F()
849 auto in_flat = GetInput(0).flat<float>(); in TEST_F()
1246 auto grad_flat = GetInput(0).flat<float>(); in TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Ddensify_test.cc58 std::vector<T> GetInput() { return ExtractVector<T>(input_); } in GetInput() function in tflite::__anon980db9da0111::DensifyOpModel
76 EXPECT_THAT(m.GetInput(), ElementsAreArray(sparse_values)); in TEST()
90 EXPECT_THAT(m.GetInput(), ElementsAreArray(sparse_values)); in TEST()
104 EXPECT_THAT(m.GetInput(), ElementsAreArray(sparse_values)); in TEST()
Dcumsum.cc37 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
38 const TfLiteTensor* axis = GetInput(context, node, kAxisTensor); in Prepare()
56 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
57 const TfLiteTensor* axis_tensor = GetInput(context, node, kAxisTensor); in Eval()
Drandom_uniform.cc67 const TfLiteTensor* input = tflite::GetInput(context, node, 0); in Prepare()
84 const TfLiteTensor* input = tflite::GetInput(context, node, 0); in EvalFloat()
129 const TfLiteTensor* input = tflite::GetInput(context, node, 0); in EvalInt()
134 int64_t min_value = IntValueFromTensor(tflite::GetInput(context, node, 1)); in EvalInt()
135 int64_t max_value = IntValueFromTensor(tflite::GetInput(context, node, 2)); in EvalInt()
Dcomplex_support.cc35 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
64 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in EvalReal()
96 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in EvalImag()
128 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in EvalAbs()
Dmultinomial.cc145 const TfLiteTensor* logits_input = tflite::GetInput(context, node, 0); in Prepare()
150 const TfLiteTensor* num_samples_input = tflite::GetInput(context, node, 1); in Prepare()
174 const TfLiteTensor* logits = tflite::GetInput(context, node, 0); in Eval()
178 const TfLiteTensor* num_samples_input = tflite::GetInput(context, node, 1); in Eval()
Done_hot.cc39 indices = GetInput(context, node, kIndicesTensor); in OneHotContext()
40 depth = GetInput(context, node, kDepthTensor); in OneHotContext()
41 on_value = GetInput(context, node, kOnValueTensor); in OneHotContext()
42 off_value = GetInput(context, node, kOffValueTensor); in OneHotContext()
Dbroadcast_to.cc38 input = GetInput(context, node, kInputTensor); in BroadcastToContext()
39 shape = GetInput(context, node, kShapeTensor); in BroadcastToContext()
93 (NumDimensions(GetInput(context, node, 0)) <= kMaxDims), in Prepare()
Dstrided_slice.cc53 input = GetInput(context, node, kInputTensor); in StridedSliceContext()
54 begin = GetInput(context, node, kBeginTensor); in StridedSliceContext()
55 end = GetInput(context, node, kEndTensor); in StridedSliceContext()
56 strides = GetInput(context, node, kStridesTensor); in StridedSliceContext()
Dreshape.cc75 const TfLiteTensor* shape = GetInput(context, node, kShapeTensor); in GetOutputShapeFromTensor()
109 const TfLiteTensor* shape = GetInput(context, node, kShapeTensor); in ShapeIsVector()
135 IsConstantTensor(GetInput(context, node, kShapeTensor))) { in Prepare()
Dbatch_to_space_nd.cc38 input = GetInput(context, node, 0); in BatchToSpaceNDContext()
39 block_shape = GetInput(context, node, 1); in BatchToSpaceNDContext()
40 crops = GetInput(context, node, 2); in BatchToSpaceNDContext()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dngrams.cc92 const TfLiteTensor* input_values = GetInput(context, node, kValues); in Prepare()
110 const TfLiteTensor* input_row_splits = GetInput(context, node, i); in Prepare()
133 const TfLiteTensor* input_values = GetInput(context, node, kValues); in Eval()
152 GetInput(context, node, kRowSplitsStart + index); in Eval()
161 GetInput(context, node, kRowSplitsStart + index); in Eval()
/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_unpooling_2d.cc63 const TfLiteTensor* input = GetInput(context, node, kDataInputTensor); in Prepare()
65 const TfLiteTensor* indices = GetInput(context, node, kIndicesTensor); in Prepare()
110 const TfLiteTensor* input = GetInput(context, node, kDataInputTensor); in Eval()
112 const TfLiteTensor* indices = GetInput(context, node, kIndicesTensor); in Eval()
Ddense_image_warp.cc94 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
96 const TfLiteTensor* flow = GetInput(context, node, kFlowTensor); in Prepare()
127 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
129 const TfLiteTensor* flow = GetInput(context, node, kFlowTensor); in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dstrided_slice.cc41 input = GetInput(context, node, kInputTensor); in StridedSliceContext()
42 begin = GetInput(context, node, kBeginTensor); in StridedSliceContext()
43 end = GetInput(context, node, kEndTensor); in StridedSliceContext()
44 strides = GetInput(context, node, kStridesTensor); in StridedSliceContext()
Dconv_common.cc96 const TfLiteTensor* input = GetInput(context, node, kConvInputTensor); in CalculateOpDataConv()
98 const TfLiteTensor* filter = GetInput(context, node, kConvWeightsTensor); in CalculateOpDataConv()
136 const TfLiteTensor* input = GetInput(context, node, kConvInputTensor); in ConvPrepare()
138 const TfLiteTensor* filter = GetInput(context, node, kConvWeightsTensor); in ConvPrepare()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc537 GetInput(context, node, ops::builtin::lstm::full::kInputTensor); in lstm_eval()
541 const TfLiteTensor* input_to_forget_weights = GetInput( in lstm_eval()
543 const TfLiteTensor* input_to_cell_weights = GetInput( in lstm_eval()
545 const TfLiteTensor* input_to_output_weights = GetInput( in lstm_eval()
550 const TfLiteTensor* recurrent_to_forget_weights = GetInput( in lstm_eval()
552 const TfLiteTensor* recurrent_to_cell_weights = GetInput( in lstm_eval()
554 const TfLiteTensor* recurrent_to_output_weights = GetInput( in lstm_eval()
580 GetInput(context, node, ops::builtin::lstm::full::kForgetGateBiasTensor); in lstm_eval()
582 GetInput(context, node, ops::builtin::lstm::full::kCellGateBiasTensor); in lstm_eval()
584 GetInput(context, node, ops::builtin::lstm::full::kOutputGateBiasTensor); in lstm_eval()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc179 IsConstantTensor(GetInput(context, node, 1)); in IsNodeSupportedByHexagon()
184 !IsConstantTensor(GetInput(context, node, 1))) in IsNodeSupportedByHexagon()
199 IsConstantTensor(GetInput(context, node, 1))); in IsNodeSupportedByHexagon()
342 IsConstantTensor(GetInput(context, node, 1)); in IsNodeSupportedByHexagon()
364 IsConstantTensor(GetInput(context, node, 0)); in IsNodeSupportedByHexagon()
369 !IsConstantTensor(GetInput(context, node, 1))) { in IsNodeSupportedByHexagon()
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test_util.cc263 a0 = GetInput(context, node, 0); in FakeFusedRegistration()
264 a1 = GetInput(context, node, 1); in FakeFusedRegistration()
266 a0 = GetInput(context, node, 0); in FakeFusedRegistration()
313 input1 = GetInput(context, node, 0); in FakeFusedRegistration()
314 input2 = GetInput(context, node, 1); in FakeFusedRegistration()
316 input1 = GetInput(context, node, 0); in FakeFusedRegistration()
478 const TfLiteTensor* input_tensor = GetInput(context, node, i); in FakeFusedRegistration()
501 const TfLiteTensor* input = GetInput(context, node, 0); in FakeFusedRegistration()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dutil.cc81 return IsBroadcastable(GetInput(context, node, 0), in IsBinaryOpSupported()
82 GetInput(context, node, 1)); in IsBinaryOpSupported()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dmul.cc51 const TfLiteTensor* input1 = GetInput(context, node, kInput1Tensor); in CalculateOpData()
52 const TfLiteTensor* input2 = GetInput(context, node, kInput2Tensor); in CalculateOpData()
81 const TfLiteTensor* input1 = GetInput(context, node, kInput1Tensor); in Prepare()
82 const TfLiteTensor* input2 = GetInput(context, node, kInput2Tensor); in Prepare()
/external/deqp/external/openglcts/modules/gles31/
Des31cShaderStorageBufferObjectTests.cpp1201 virtual const char* GetInput(std::vector<GLubyte>& in_data) = 0;
1216 const char* glsl_vs = GetInput(in_data); in Run()
1281 virtual const char* GetInput(std::vector<GLubyte>& in_data) = 0;
1294 ss << "layout(local_size_x = 1) in;\n" << GetInput(in_data); in Run()
1371 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in glcts::__anonb63440200111::BasicStd430LayoutCase1VS
1379 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in glcts::__anonb63440200111::BasicStd430LayoutCase1CS
1418 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in glcts::__anonb63440200111::BasicStd430LayoutCase2VS
1426 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in glcts::__anonb63440200111::BasicStd430LayoutCase2CS
1464 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in glcts::__anonb63440200111::BasicStd430LayoutCase3VS
1472 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in glcts::__anonb63440200111::BasicStd430LayoutCase3CS
[all …]
/external/deqp/external/openglcts/modules/gl/
Dgl4cES31CompatibilityShaderStorageBufferObjectTests.cpp1201 virtual const char* GetInput(std::vector<GLubyte>& in_data) = 0;
1216 const char* glsl_vs = GetInput(in_data); in Run()
1281 virtual const char* GetInput(std::vector<GLubyte>& in_data) = 0;
1294 ss << "layout(local_size_x = 1) in;\n" << GetInput(in_data); in Run()
1372 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::es31compatibility::__anonb97320c50111::BasicStd430LayoutCase1VS
1380 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::es31compatibility::__anonb97320c50111::BasicStd430LayoutCase1CS
1419 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::es31compatibility::__anonb97320c50111::BasicStd430LayoutCase2VS
1427 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::es31compatibility::__anonb97320c50111::BasicStd430LayoutCase2CS
1464 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::es31compatibility::__anonb97320c50111::BasicStd430LayoutCase3VS
1472 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::es31compatibility::__anonb97320c50111::BasicStd430LayoutCase3CS
[all …]
Dgl4cShaderStorageBufferObjectTests.cpp1265 virtual const char* GetInput(std::vector<GLubyte>& in_data) = 0;
1280 const char* glsl_vs = GetInput(in_data); in Run()
1341 virtual const char* GetInput(std::vector<GLubyte>& in_data) = 0;
1354 ss << "layout(local_size_x = 1) in;\n" << GetInput(in_data); in Run()
1429 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::__anon69e0890f0111::BasicStd430LayoutCase1VS
1437 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::__anon69e0890f0111::BasicStd430LayoutCase1CS
1476 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::__anon69e0890f0111::BasicStd430LayoutCase2VS
1484 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::__anon69e0890f0111::BasicStd430LayoutCase2CS
1522 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::__anon69e0890f0111::BasicStd430LayoutCase3VS
1530 virtual const char* GetInput(std::vector<GLubyte>& in_data) in GetInput() function in gl4cts::__anon69e0890f0111::BasicStd430LayoutCase3CS
[all …]

12345