Home
last modified time | relevance | path

Searched refs:tensor_2 (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/
Dmodify_model_interface_test.cc100 auto tensor_2 = absl::make_unique<TensorT>(); in CreateQuantizedModelSingleInputOutput() local
101 tensor_2->quantization = absl::make_unique<QuantizationParametersT>(); in CreateQuantizedModelSingleInputOutput()
102 tensor_2->quantization->scale.push_back(0.12); in CreateQuantizedModelSingleInputOutput()
103 tensor_2->quantization->zero_point.push_back(50); in CreateQuantizedModelSingleInputOutput()
104 tensor_2->name = "tensor_2"; in CreateQuantizedModelSingleInputOutput()
105 tensor_2->shape = {}; in CreateQuantizedModelSingleInputOutput()
106 tensor_2->type = quantization_type; in CreateQuantizedModelSingleInputOutput()
115 model->subgraphs[0]->tensors.push_back(std::move(tensor_2)); in CreateQuantizedModelSingleInputOutput()
202 auto tensor_2 = absl::make_unique<TensorT>(); in CreateQuantizedModelMultipleInputOutput() local
203 tensor_2->quantization = absl::make_unique<QuantizationParametersT>(); in CreateQuantizedModelMultipleInputOutput()
[all …]
Dquantize_model.cc981 TensorT* tensor_2 = subgraph->tensors[op->outputs[index_2]].get(); in QuantizeSharedRange() local
983 tensor_2->quantization->min[0]); in QuantizeSharedRange()
985 tensor_2->quantization->max[0]); in QuantizeSharedRange()
999 for (TensorT* tensor : {tensor_1, tensor_2}) { in QuantizeSharedRange()
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_v2_ops_test.py439 tensor_2, tensor_id_2 = debug_summary(c)
440 self.assertAllEqual(tensor_1, tensor_2)
447 tensor_2, tensor_id_2 = debug_summary(c)
448 self.assertAllEqual(tensor_1, tensor_2)
453 tensor_2, tensor_id_2 = debug_summary(c)
454 self.assertAllEqual(tensor_1, tensor_2)
539 tensor_2, tensor_id_2 = debug_summary(c)
540 self.assertAllEqual(tensor_1, tensor_2)
545 tensor_2, tensor_id_2 = debug_summary(c)
546 self.assertAllEqual(tensor_1, tensor_2)
[all …]
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_v1_test.py427 tensor_2 = sparse_tensor.SparseTensor([[0, 0]], [2], [1, 1])
429 tensor_1, tensor_2)
437 tensor_2 = sparse_tensor.SparseTensorValue([[0, 0]], [2], [1, 1])
439 tensor_1, tensor_2)
447 tensor_2 = ragged_tensor.RaggedTensor.from_row_splits(
450 tensor_1, tensor_2)
459 tensor_2 = ragged_tensor_value.RaggedTensorValue(
462 tensor_1, tensor_2)
/external/tensorflow/tensorflow/core/ops/
Ddata_flow_ops_test.cc141 Tensor tensor_2 = test::AsTensor<int32>( in TEST() local
152 op.input_tensors[0] = &tensor_2; in TEST()
160 tensor_2.flat<int32>()(3) = 10000; in TEST()