Searched refs:tensor_b (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | resource_variable_test.cc | 97 TfLiteTensor tensor_a, tensor_b; in TEST() local
101 InitTensor(shape_b, kTfLiteDynamic, 4.0, &tensor_b); in TEST()
115 EXPECT_EQ(kTfLiteOk, var.AssignFrom(&tensor_b)); in TEST()
128 TfLiteTensorFree(&tensor_b); in TEST()
138 TfLiteTensor tensor_a, tensor_b; in TEST() local
142 InitTensor(shape_b, kTfLiteDynamic, 4.0, &tensor_b); in TEST()
156 EXPECT_EQ(kTfLiteOk, var.AssignFrom(&tensor_b)); in TEST()
169 TfLiteTensorFree(&tensor_b); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | squared_difference.cc | 45 const auto& tensor_b = context->tensors[tensor_b_index]; in PopulateSubGraph() local
50 TF_LITE_ENSURE_STATUS(ComputeAndAddMinAndMax(context, tensor_b)); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | benchmarks_test.py | 233 tensor_b = constant_op.constant(24.0)
234 self._benchmark_add(tensor_a, tensor_b)
238 tensor_b = constant_op.constant(24.0)
239 self._benchmark_add_operator_overload(tensor_a, tensor_b)
243 tensor_b = constant_op.constant(24)
244 self._benchmark_add(tensor_a, tensor_b)
248 tensor_b = constant_op.constant([[24.0, 24.0], [24.0, 24.0]])
249 self._benchmark_add(tensor_a, tensor_b)
253 tensor_b = constant_op.constant([[24, 24], [24, 24]])
254 self._benchmark_add(tensor_a, tensor_b)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5533 nvinfer1::ITensor* tensor_b = prepare_matmul_operand(input_b, &transpose_b); in ConvertMatMulHelper() local
5537 *tensor_a, get_matrix_op(tensor_a, transpose_a), *tensor_b, in ConvertMatMulHelper()
5538 get_matrix_op(tensor_b, transpose_b)); in ConvertMatMulHelper()
|