| /external/XNNPACK/test/ |
| D | subgraph-nchw.cc | 14 .add_tensor({1, 256, 256, 3}, kDynamic, 0) in TEST()
15 .add_tensor({32, 3, 3, 3}, kStaticDense, 1) in TEST()
16 .add_tensor({32}, kStaticDense, 2) in TEST()
17 .add_tensor({1, 128, 128, 32}, kDynamic, 3) in TEST()
29 .add_tensor({1, 256, 256, 3}, kDynamic, 0) in TEST()
30 .add_tensor({32, 3, 3, 3}, kStaticDense, 1) in TEST()
31 .add_tensor({32}, kStaticDense, 2) in TEST()
32 .add_tensor({1, 128, 128, 32}, kDynamic, 3) in TEST()
33 .add_tensor({32}, kDynamic, 4) in TEST()
47 .add_tensor({1, 256, 256, 3}, kDynamic, 0) in TEST()
[all …]
|
| D | subgraph-tester.h | 43 inline SubgraphTester& add_tensor(const std::vector<size_t>& dims, in add_tensor() function
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | fuse_add_to_conv_test.cc | 55 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
56 add_tensor.shape = Linear(16); in TEST()
57 add_tensor.data.resize(16); in TEST()
59 add_attr.param = add_tensor; in TEST()
98 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
99 add_tensor.shape = Linear(2); in TEST()
100 add_tensor.data = {0.3f, 0.7f}; in TEST()
102 add_attr.param = add_tensor; in TEST()
119 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
120 add_tensor.shape = Linear(4); in TEST()
[all …]
|
| D | add_quant_adjustments_test.cc | 57 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
58 add_tensor.shape = Linear(8); in TEST()
59 add_tensor.data.resize(8); in TEST()
61 add_attr.param = add_tensor; in TEST()
101 Tensor<Linear, DataType::FLOAT32> add_tensor; in TEST() local
102 add_tensor.shape = Linear(8); in TEST()
103 add_tensor.data.resize(8); in TEST()
105 add_attr.param = add_tensor; in TEST()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_tensor_data.h | 78 Tensor* add_tensor(TensorConstructorArgs&&... args);
129 Tensor* VariantTensorData::add_tensor(TensorConstructorArgs&&... args) { in add_tensor() function
|
| D | extension_type_variant.cc | 29 data->add_tensor(tensor); in Encode()
|
| D | attr_value_util.cc | 559 v.AsProtoTensorContent(out->mutable_list()->add_tensor()); in SetAttrValue()
561 v.AsProtoField(out->mutable_list()->add_tensor()); in SetAttrValue()
573 *out->mutable_list()->add_tensor() = v; in SetAttrValue()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | optional_test.py | 187 add_tensor = math_ops.add_n(
189 add_opt = optional_ops._OptionalImpl(add_tensor, opt1.element_spec)
195 add_tensor = math_ops.add_n(
197 add_opt = optional_ops._OptionalImpl(add_tensor, opt_none1.element_spec)
212 add_tensor = math_ops.add_n(
214 add_opt = optional_ops._OptionalImpl(add_tensor, opt3.element_spec)
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 143 const Tensor& add_tensor = MklGetInput(ctx, kInputIndex_Add); in Compute() local
172 static_cast<void*>(const_cast<T*>(add_tensor.flat<T>().data())); in Compute()
179 memory::desc({add_tensor.NumElements()}, MklDnnType<T>(), in Compute()
|
| D | mkl_conv_ops.cc | 877 const Tensor& add_tensor = MklGetInput(context, kInputIndex_Add); in AllocateOutputTensor() local
908 const_cast<Toutput*>(add_tensor.flat<Toutput>().data())); in AllocateOutputTensor()
915 memory::desc({add_tensor.NumElements()}, MklDnnType<Toutput>(), in AllocateOutputTensor()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_writer.h | 129 SavedSliceMeta* ssm = sts_.mutable_meta()->add_tensor(); in Add()
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_ref.cc | 169 NamedTensorProto* tproto = resp->add_tensor(); in RecordRun()
302 NamedTensorProto* tproto = resp->add_tensor(); in RecordPRun()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 1351 add_tensor = a_tensor + b_tensor
1358 self.assertEqual(mon_sess.run(fetches=add_tensor), [15])
1369 add_tensor = a_tensor + b_tensor + c_tensor
1378 mon_sess.run(fetches=add_tensor, feed_dict=feed_dict), [35])
1390 add_tensor = a_tensor + b_tensor
1398 mon_sess.run(fetches=add_tensor)
1408 add_tensor = a_tensor + b_tensor
1416 mon_sess.run(fetches=add_tensor, feed_dict={b_tensor: [10]})
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 505 [response] { return response->add_tensor(); }, in RunComponentFunction()
544 retvals.data(), [queue_response] { return queue_response->add_tensor(); }, in ExecuteOp()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | message_wrappers.cc | 882 NamedTensorProto* response_tensor = response_.add_tensor(); in AddTensorFromRunGraphResponse()
934 NamedTensorProto* response_tensor = response_->add_tensor(); in AddTensorFromRunGraphResponse()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | export_utils.cc | 210 *list->add_tensor() = tensor; in ConvertAttribute()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | snapshot_util.cc | 205 TensorProto* t = record.add_tensor(); in WriteTensors()
|