Home
last modified time | relevance | path

Searched refs:input_tensor1 (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/calibrator/integration_test/
Dcustom_aggregator_op_test.py51 input_tensor1 = array_ops.constant([1.0, 2.0, 3.0, 4.0, 5.0],
54 input_tensor1, '2')
72 input_tensor1 = array_ops.constant([1.0, 2.0, 3.0, 4.0, 5.0],
75 input_tensor1, '4')
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dfully_connected.py141 input_tensor1 = tf.compat.v1.placeholder(
149 input_tensors = [input_tensor1]
153 input_tensors = [input_tensor1, input_tensor2]
156 input_tensor1,
Dunroll_batch_matmul.py57 input_tensor1 = tf.compat.v1.placeholder(
63 input_tensor1,
67 return [input_tensor1, input_tensor2], [out]
Deye.py40 input_tensor1 = tf.compat.v1.placeholder(
45 num_columns=input_tensor1,
48 return [input_tensor0, input_tensor1], [outs]
/external/tensorflow/tensorflow/python/ops/
Dcollective_ops_xla_test.py59 input_tensor1 = array_ops.identity(f(constant))
62 input_tensor1, group_size, group_key, instance_key1, 'Add', 'Id')
Dcollective_ops_test.py553 input_tensor1 = array_ops.identity(constant)
556 input_tensor1, group_size, group_key, instance_key1, 'Add',
/external/pytorch/test/export/
Dtest_pass_infra.py131 input_tensor1 = torch.tensor(5.0)
134 ep_before = torch.export.export(my_module, (input_tensor1, input_tensor2))
/external/tensorflow/tensorflow/lite/python/metrics/
Dmetrics_nonportable_test.py562 input_tensor1 = tf.keras.layers.Input(
566 output = tf.keras.layers.Add()([input_tensor1, input_tensor2])
568 inputs=[input_tensor1, input_tensor2], outputs=output)
/external/tensorflow/tensorflow/lite/python/
Dmetrics_nonportable_test.py561 input_tensor1 = tf.keras.layers.Input(
565 output = tf.keras.layers.Add()([input_tensor1, input_tensor2])
567 inputs=[input_tensor1, input_tensor2], outputs=output)
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc1060 int input_tensor1 = 1; in Parse() local
1072 input_tensor1 = 0; in Parse()
1077 RETURN_IF_ERROR(reader->AddInput(node, input_tensor1)); in Parse()
/external/pytorch/test/dynamo/
Dtest_misc.py2810 input_tensor1 = torch.randn(vec1)
2813 args = {"input": input_tensor1, "vec2": input_tensor2, "out": out_tensor}
/external/pytorch/torch/testing/_internal/
Dcommon_methods_invocations.py4703 input_tensor1 = create_tensor(batch_shape + [in_feat1])
4707 yield SampleInput(input_tensor1, input_tensor2, weight)
4710 yield SampleInput(input_tensor1, input_tensor2, weight, bias)