Searched refs:tensor_a (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
100 InitTensor(shape_a, kTfLiteDynamic, 1.0, &tensor_a); in TEST()
103 EXPECT_EQ(kTfLiteOk, var.AssignFrom(&tensor_a)); in TEST()
127 TfLiteTensorFree(&tensor_a); in TEST()
138 TfLiteTensor tensor_a, tensor_b; in TEST() local
141 InitTensor(shape_a, kTfLiteDynamic, 1.0, &tensor_a); in TEST()
144 EXPECT_EQ(kTfLiteOk, var.AssignFrom(&tensor_a)); in TEST()
168 TfLiteTensorFree(&tensor_a); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | squared_difference.cc | 44 const auto& tensor_a = context->tensors[tensor_a_index]; in PopulateSubGraph() local
49 TF_LITE_ENSURE_STATUS(ComputeAndAddMinAndMax(context, tensor_a)); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | benchmarks_test.py | 232 tensor_a = constant_op.constant(42.0)
234 self._benchmark_add(tensor_a, tensor_b)
237 tensor_a = constant_op.constant(42.0)
239 self._benchmark_add_operator_overload(tensor_a, tensor_b)
242 tensor_a = constant_op.constant(42)
244 self._benchmark_add(tensor_a, tensor_b)
247 tensor_a = constant_op.constant([[42.0, 42.0], [42.0, 42.0]])
249 self._benchmark_add(tensor_a, tensor_b)
252 tensor_a = constant_op.constant([[42, 42], [42, 42]])
254 self._benchmark_add(tensor_a, tensor_b)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5418 nvinfer1::ITensor* tensor_a, in ConvertFullyConnectedHelper() argument
5422 auto input_dim = tensor_a->getDimensions(); in ConvertFullyConnectedHelper()
5427 params->converter, TRT_TensorOrWeights(tensor_a), input_dim, in ConvertFullyConnectedHelper()
5428 /*validation_only=*/false, &tensor_a, node_def, /*op_instance=*/0)); in ConvertFullyConnectedHelper()
5442 *tensor_a, noutput, weights.GetTrtWeights(), biases.GetTrtWeights()); in ConvertFullyConnectedHelper()
5532 nvinfer1::ITensor* tensor_a = prepare_matmul_operand(input_a, &transpose_a); in ConvertMatMulHelper() local
5537 *tensor_a, get_matrix_op(tensor_a, transpose_a), *tensor_b, in ConvertMatMulHelper()
|