/external/tensorflow/tensorflow/core/kernels/ |
D | parse_tensor_test.cc | 84 test::ExpectTensorEqual<Eigen::half>(parse_output, GetInput(0)); in TEST_F() 93 test::ExpectTensorEqual<float>(parse_output, GetInput(0)); in TEST_F() 102 test::ExpectTensorEqual<double>(parse_output, GetInput(0)); in TEST_F() 111 test::ExpectTensorEqual<int64>(parse_output, GetInput(0)); in TEST_F() 120 test::ExpectTensorEqual<int32>(parse_output, GetInput(0)); in TEST_F() 129 test::ExpectTensorEqual<int16>(parse_output, GetInput(0)); in TEST_F() 138 test::ExpectTensorEqual<int8>(parse_output, GetInput(0)); in TEST_F() 147 test::ExpectTensorEqual<uint16>(parse_output, GetInput(0)); in TEST_F() 156 test::ExpectTensorEqual<uint8>(parse_output, GetInput(0)); in TEST_F() 166 test::ExpectTensorEqual<complex64>(parse_output, GetInput(0)); in TEST_F() [all …]
|
D | fake_quant_ops_test.cc | 379 auto input_flat = GetInput(0).flat<float>(); in TEST_F() 409 auto input_flat = GetInput(0).flat<float>(); in TEST_F() 440 auto input_flat = GetInput(0).flat<float>(); in TEST_F() 471 auto input_flat = GetInput(0).flat<float>(); in TEST_F() 669 auto in_flat = GetInput(0).flat<float>(); in TEST_F() 714 auto in_flat = GetInput(0).flat<float>(); in TEST_F() 758 auto in_flat = GetInput(0).flat<float>(); in TEST_F() 803 auto in_flat = GetInput(0).flat<float>(); in TEST_F() 848 auto in_flat = GetInput(0).flat<float>(); in TEST_F() 1245 auto grad_flat = GetInput(0).flat<float>(); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm.cc | 138 GetInput(context, node, kInputToForgetWeightsTensor); in CheckInputTensorDimensions() 144 GetInput(context, node, kInputToCellWeightsTensor); in CheckInputTensorDimensions() 160 GetInput(context, node, kRecurrentToForgetWeightsTensor); in CheckInputTensorDimensions() 168 GetInput(context, node, kRecurrentToCellWeightsTensor); in CheckInputTensorDimensions() 225 GetInput(context, node, kForgetGateBiasTensor); in CheckInputTensorDimensions() 229 const TfLiteTensor* cell_bias = GetInput(context, node, kCellGateBiasTensor); in CheckInputTensorDimensions() 234 GetInput(context, node, kOutputGateBiasTensor); in CheckInputTensorDimensions() 275 GetInput(context, node, kForgetLayerNormCoefficientsTensor); in CheckInputTensorDimensions() 282 GetInput(context, node, kCellLayerNormCoefficientsTensor); in CheckInputTensorDimensions() 289 GetInput(context, node, kOutputLayerNormCoefficientsTensor); in CheckInputTensorDimensions() [all …]
|
D | embedding_lookup_sparse.cc | 84 const TfLiteTensor* ids = GetInput(context, node, 0); in Prepare() 88 const TfLiteTensor* indices = GetInput(context, node, 1); in Prepare() 92 const TfLiteTensor* shape = GetInput(context, node, 2); in Prepare() 96 const TfLiteTensor* weights = GetInput(context, node, 3); in Prepare() 105 const TfLiteTensor* value = GetInput(context, node, 4); in Prepare() 142 const TfLiteTensor* ids = GetInput(context, node, 0); in Eval() 143 const TfLiteTensor* indices = GetInput(context, node, 1); in Eval() 144 const TfLiteTensor* dense_shape = GetInput(context, node, 2); in Eval() 145 const TfLiteTensor* weights = GetInput(context, node, 3); in Eval() 146 const TfLiteTensor* value = GetInput(context, node, 4); in Eval()
|
D | comparisons.cc | 36 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in ComparisonPrepare() 37 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in ComparisonPrepare() 127 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in EqualEval() 128 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in EqualEval() 163 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in NotEqualEval() 164 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in NotEqualEval() 198 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in GreaterEval() 199 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in GreaterEval() 230 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in GreaterEqualEval() 231 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in GreaterEqualEval() [all …]
|
D | unidirectional_sequence_lstm.cc | 120 GetInput(context, node, kInputToForgetWeightsTensor); in CheckInputTensorDimensions() 126 GetInput(context, node, kInputToCellWeightsTensor); in CheckInputTensorDimensions() 142 GetInput(context, node, kRecurrentToForgetWeightsTensor); in CheckInputTensorDimensions() 150 GetInput(context, node, kRecurrentToCellWeightsTensor); in CheckInputTensorDimensions() 208 GetInput(context, node, kForgetGateBiasTensor); in CheckInputTensorDimensions() 212 const TfLiteTensor* cell_bias = GetInput(context, node, kCellGateBiasTensor); in CheckInputTensorDimensions() 217 GetInput(context, node, kOutputGateBiasTensor); in CheckInputTensorDimensions() 260 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare() 271 GetInput(context, node, kInputToOutputWeightsTensor); in Prepare() 277 GetInput(context, node, kRecurrentToOutputWeightsTensor); in Prepare() [all …]
|
D | bidirectional_sequence_lstm.cc | 179 GetInput(context, node, input_to_forget_weights_tensor); in CheckLstmTensorDimensionsAndTypes() 197 GetInput(context, node, input_to_cell_weights_tensor); in CheckLstmTensorDimensionsAndTypes() 205 GetInput(context, node, input_to_output_weights_tensor); in CheckLstmTensorDimensionsAndTypes() 225 GetInput(context, node, recurrent_to_forget_weights_tensor); in CheckLstmTensorDimensionsAndTypes() 235 GetInput(context, node, recurrent_to_cell_weights_tensor); in CheckLstmTensorDimensionsAndTypes() 302 GetInput(context, node, forget_gate_bias_tensor); in CheckLstmTensorDimensionsAndTypes() 308 GetInput(context, node, cell_gate_bias_tensor); in CheckLstmTensorDimensionsAndTypes() 314 GetInput(context, node, output_gate_bias_tensor); in CheckLstmTensorDimensionsAndTypes() 398 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare() 407 GetInput(context, node, kFwInputToOutputWeightsTensor); in Prepare() [all …]
|
D | sparse_to_dense.cc | 150 const TfLiteTensor* indices = GetInput(context, node, kIndicesTensor); in Prepare() 152 GetInput(context, node, kOutputShapeTensor); in Prepare() 153 const TfLiteTensor* values = GetInput(context, node, kValueInputTensor); in Prepare() 155 GetInput(context, node, kDefaultValueTensor); in Prepare() 192 const TfLiteTensor* indices = GetInput(context, node, kIndicesTensor); in SparseToDenseImpl() 194 GetInput(context, node, kOutputShapeTensor); in SparseToDenseImpl() 195 const TfLiteTensor* values = GetInput(context, node, kValueInputTensor); in SparseToDenseImpl() 197 GetInput(context, node, kDefaultValueTensor); in SparseToDenseImpl() 220 const TfLiteTensor* indices = GetInput(context, node, kIndicesTensor); in Eval() 221 const TfLiteTensor* values = GetInput(context, node, kValueInputTensor); in Eval()
|
D | select.cc | 37 GetInput(context, node, kInputTensorCondition); in SelectPrepare() 38 const TfLiteTensor* input_x = GetInput(context, node, kInputTensorX); in SelectPrepare() 39 const TfLiteTensor* input_y = GetInput(context, node, kInputTensorY); in SelectPrepare() 66 GetInput(context, node, kInputTensorCondition); in SelectEval() 67 const TfLiteTensor* input_x = GetInput(context, node, kInputTensorX); in SelectEval() 68 const TfLiteTensor* input_y = GetInput(context, node, kInputTensorY); in SelectEval()
|
D | basic_rnn.cc | 64 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare() 65 const TfLiteTensor* input_weights = GetInput(context, node, kWeightsTensor); in Prepare() 67 GetInput(context, node, kRecurrentWeightsTensor); in Prepare() 68 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Prepare() 70 GetInput(context, node, kHiddenStateTensor); in Prepare() 221 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval() 222 const TfLiteTensor* input_weights = GetInput(context, node, kWeightsTensor); in Eval() 224 GetInput(context, node, kRecurrentWeightsTensor); in Eval() 225 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Eval()
|
D | hashtable_lookup.cc | 62 const TfLiteTensor* lookup = GetInput(context, node, 0); in Prepare() 66 const TfLiteTensor* key = GetInput(context, node, 1); in Prepare() 70 const TfLiteTensor* value = GetInput(context, node, 2); in Prepare() 104 const TfLiteTensor* lookup = GetInput(context, node, 0); in Eval() 105 const TfLiteTensor* key = GetInput(context, node, 1); in Eval() 106 const TfLiteTensor* value = GetInput(context, node, 2); in Eval()
|
D | lsh_projection.cc | 79 const TfLiteTensor* hash = GetInput(context, node, 0); in Resize() 84 const TfLiteTensor* input = GetInput(context, node, 1); in Resize() 88 const TfLiteTensor* weight = GetInput(context, node, 2); in Resize() 175 const TfLiteTensor* hash = GetInput(context, node, 0); in Eval() 176 const TfLiteTensor* input = GetInput(context, node, 1); in Eval() 178 NumInputs(node) == 2 ? nullptr : GetInput(context, node, 2); in Eval()
|
D | range.cc | 80 const TfLiteTensor* start = GetInput(context, node, kStartTensor); in Prepare() 81 const TfLiteTensor* limit = GetInput(context, node, kLimitTensor); in Prepare() 82 const TfLiteTensor* delta = GetInput(context, node, kDeltaTensor); in Prepare() 133 const TfLiteTensor* start = GetInput(context, node, kStartTensor); in Eval() 134 const TfLiteTensor* limit = GetInput(context, node, kLimitTensor); in Eval() 135 const TfLiteTensor* delta = GetInput(context, node, kDeltaTensor); in Eval()
|
D | unidirectional_sequence_rnn.cc | 59 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare() 60 const TfLiteTensor* input_weights = GetInput(context, node, kWeightsTensor); in Prepare() 62 GetInput(context, node, kRecurrentWeightsTensor); in Prepare() 63 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Prepare() 65 GetInput(context, node, kHiddenStateTensor); in Prepare() 291 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval() 292 const TfLiteTensor* input_weights = GetInput(context, node, kWeightsTensor); in Eval() 294 GetInput(context, node, kRecurrentWeightsTensor); in Eval() 295 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Eval() 298 const_cast<TfLiteTensor*>(GetInput(context, node, kHiddenStateTensor)); in Eval()
|
D | reshape.cc | 33 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in ResizeOutput() 64 const TfLiteTensor* shape = GetInput(context, node, kShapeTensor); in GetOutputShapeFromTensor() 97 const TfLiteTensor* shape = GetInput(context, node, kShapeTensor); in ShapeIsVector() 120 IsConstantTensor(GetInput(context, node, kShapeTensor))) { in Prepare() 131 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
|
D | tile.cc | 47 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in ResizeOutput() 49 const TfLiteTensor* multipliers = GetInput(context, node, kInputMultipliers); in ResizeOutput() 141 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare() 146 const TfLiteTensor* multipliers = GetInput(context, node, kInputMultipliers); in Prepare() 164 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval() 166 const TfLiteTensor* multipliers = GetInput(context, node, kInputMultipliers); in Eval()
|
D | reverse_sequence.cc | 36 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare() 37 const TfLiteTensor* seq_lengths = GetInput(context, node, kSeqLengthsTensor); in Prepare() 65 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in ReverseSequenceImpl() 67 GetInput(context, node, kSeqLengthsTensor); in ReverseSequenceImpl() 99 GetInput(context, node, kSeqLengthsTensor); in ReverseSequenceHelper()
|
D | bidirectional_sequence_rnn.cc | 96 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare() 98 GetInput(context, node, kFwWeightsTensor); in Prepare() 100 GetInput(context, node, kFwRecurrentWeightsTensor); in Prepare() 101 const TfLiteTensor* fw_bias = GetInput(context, node, kFwBiasTensor); in Prepare() 103 GetInput(context, node, kFwHiddenStateTensor); in Prepare() 105 GetInput(context, node, kBwWeightsTensor); in Prepare() 107 GetInput(context, node, kBwRecurrentWeightsTensor); in Prepare() 108 const TfLiteTensor* bw_bias = GetInput(context, node, kBwBiasTensor); in Prepare() 110 GetInput(context, node, kBwHiddenStateTensor); in Prepare() 595 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval() [all …]
|
D | slice.cc | 109 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare() 110 const TfLiteTensor* begin = GetInput(context, node, kBeginTensor); in Prepare() 111 const TfLiteTensor* size = GetInput(context, node, kSizeTensor); in Prepare() 137 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval() 138 const TfLiteTensor* begin = GetInput(context, node, kBeginTensor); in Eval() 139 const TfLiteTensor* size = GetInput(context, node, kSizeTensor); in Eval()
|
D | expand_dims.cc | 75 const TfLiteTensor* input = GetInput(context, node, kInput); in Prepare() 76 const TfLiteTensor* axis = GetInput(context, node, kAxis); in Prepare() 91 const TfLiteTensor* input = GetInput(context, node, kInput); in Eval() 93 const TfLiteTensor* axis = GetInput(context, node, kAxis); in Eval()
|
D | fill.cc | 72 const TfLiteTensor* dims = GetInput(context, node, kDimsTensor); in Prepare() 73 const TfLiteTensor* value = GetInput(context, node, kValueTensor); in Prepare() 97 const TfLiteTensor* value = GetInput(context, node, kValueTensor); in Eval() 102 const TfLiteTensor* dims = GetInput(context, node, kDimsTensor); in Eval()
|
D | logical.cc | 54 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in Prepare() 55 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in Prepare() 84 const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); in LogicalImpl() 85 const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); in LogicalImpl()
|
D | transpose_conv.cc | 141 GetInput(context, node, kOutputShapeTensor); in Prepare() 142 const TfLiteTensor* weights = GetInput(context, node, kWeightsTensor); in Prepare() 143 const TfLiteTensor* input = GetInput(context, node, kDataInputTensor); in Prepare() 177 GetInput(context, node, kOutputShapeTensor); in Eval() 178 const TfLiteTensor* weights = GetInput(context, node, kWeightsTensor); in Eval() 179 const TfLiteTensor* input = GetInput(context, node, kDataInputTensor); in Eval()
|
D | reverse.cc | 36 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare() 37 const TfLiteTensor* axis = GetInput(context, node, kAxisTensor); in Prepare() 68 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval() 69 const TfLiteTensor* axis_tensor = GetInput(context, node, kAxisTensor); in Eval()
|
/external/tensorflow/tensorflow/contrib/lite/kernels/ |
D | lsh_projection.cc | 79 const TfLiteTensor* hash = GetInput(context, node, 0); in Resize() 84 const TfLiteTensor* input = GetInput(context, node, 1); in Resize() 88 const TfLiteTensor* weight = GetInput(context, node, 2); in Resize() 175 const TfLiteTensor* hash = GetInput(context, node, 0); in Eval() 176 const TfLiteTensor* input = GetInput(context, node, 1); in Eval() 178 NumInputs(node) == 2 ? nullptr : GetInput(context, node, 2); in Eval()
|