| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 81 virtual Status Run(const gtl::InlinedVector<TensorValue, 4>& inputs,
87 const gtl::InlinedVector<TensorValue, 4>& inputs,
147 Status Run(const gtl::InlinedVector<TensorValue, 4>& inputs,
152 const gtl::InlinedVector<TensorValue, 4>& inputs,
209 Status Run(const gtl::InlinedVector<TensorValue, 4>& inputs,
213 const gtl::InlinedVector<TensorValue, 4>& inputs,
|
| D | kernel_and_device_test.cc | 131 gtl::InlinedVector<TensorValue, 4> inputs; in BM_KernelAndDeviceRun()
132 inputs.push_back(TensorValue(&t)); in BM_KernelAndDeviceRun()
133 inputs.push_back(TensorValue(&t)); in BM_KernelAndDeviceRun()
|
| D | kernel_and_device.cc | 151 Status KernelAndDeviceOp::Run(const gtl::InlinedVector<TensorValue, 4>& inputs, in Run() argument
163 const gtl::InlinedVector<TensorValue, 4>& inputs, in Run() argument
207 const gtl::InlinedVector<TensorValue, 4>& inputs, in Run() argument
306 const gtl::InlinedVector<TensorValue, 4>& inputs, in Run() argument
341 for (const TensorValue& tensor_value : inputs) { in Run()
|
| D | tensor_handle.h | 82 Status TensorValue(tensorflow::TensorValue* t);
|
| D | tensor_handle.cc | 151 Status TensorHandle::TensorValue(tensorflow::TensorValue* t) { in TensorValue() function in tensorflow::TensorHandle
154 *t = tensorflow::TensorValue(&tensor_); in TensorValue()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_dataset_op_test.cc | 57 gtl::InlinedVector<TensorValue, 4> *inputs, in CreateTensorDatasetContext() argument
141 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_P()
191 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F()
217 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F()
243 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F()
269 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F()
300 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F()
326 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_P()
357 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_P()
389 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_P()
[all …]
|
| D | dataset_test_base.h | 94 gtl::InlinedVector<TensorValue, 4> range_inputs; in CreateRangeDataset()
141 const gtl::InlinedVector<TensorValue, 4>& inputs);
145 gtl::InlinedVector<TensorValue, 4>* inputs,
163 Status AddDatasetInputFromArray(gtl::InlinedVector<TensorValue, 4>* inputs, in AddDatasetInputFromArray() argument
176 Status AddDatasetInput(gtl::InlinedVector<TensorValue, 4>* inputs,
|
| D | sparse_tensor_slice_dataset_op_test.cc | 51 OpKernel *const op_kernel, gtl::InlinedVector<TensorValue, 4> *inputs, in CreateSparseTensorSliceDatasetContext() argument
137 gtl::InlinedVector<TensorValue, 4> inputs = { in TEST_F()
187 gtl::InlinedVector<TensorValue, 4> inputs = {&indices, &values, &dense_shape}; in TEST_F()
212 gtl::InlinedVector<TensorValue, 4> inputs = { in TEST_F()
245 gtl::InlinedVector<TensorValue, 4> inputs = { in TEST_F()
278 gtl::InlinedVector<TensorValue, 4> inputs = { in TEST_F()
307 gtl::InlinedVector<TensorValue, 4> inputs = {&indices, &values, &dense_shape}; in TEST_F()
337 gtl::InlinedVector<TensorValue, 4> inputs = { in TEST_F()
376 gtl::InlinedVector<TensorValue, 4> inputs = { in TEST_F()
415 gtl::InlinedVector<TensorValue, 4> inputs = {&indices, &values, &dense_shape}; in TEST_F()
[all …]
|
| D | tensor_slice_dataset_op_test.cc | 59 gtl::InlinedVector<TensorValue, 4> *inputs, in CreateTensorSliceDatasetContext() argument
142 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F()
209 gtl::InlinedVector<TensorValue, 4> inputs = {&t1, &t2}; in TEST_F()
241 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F()
284 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F()
328 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F()
361 gtl::InlinedVector<TensorValue, 4> inputs = {&t1, &t2}; in TEST_F()
399 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F()
448 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F()
493 gtl::InlinedVector<TensorValue, 4> inputs = {&t1, &t2}; in TEST_F()
[all …]
|
| D | repeat_dataset_op_test.cc | 53 OpKernel *op_kernel, gtl::InlinedVector<TensorValue, 4> *const inputs, in CreateRepeatDatasetContext() argument
140 gtl::InlinedVector<TensorValue, 4> inputs_for_repeat_dataset; in TEST_P()
212 gtl::InlinedVector<TensorValue, 4> inputs_for_repeat_dataset; in TEST_F()
242 gtl::InlinedVector<TensorValue, 4> inputs_for_repeat_dataset; in TEST_P()
272 gtl::InlinedVector<TensorValue, 4> inputs_for_repeat_dataset; in TEST_P()
302 gtl::InlinedVector<TensorValue, 4> inputs_for_repeat_dataset; in TEST_P()
332 gtl::InlinedVector<TensorValue, 4> inputs_for_repeat_dataset; in TEST_F()
367 gtl::InlinedVector<TensorValue, 4> inputs_for_repeat_dataset; in TEST_P()
404 gtl::InlinedVector<TensorValue, 4> inputs_for_repeat_dataset; in TEST_P()
441 gtl::InlinedVector<TensorValue, 4> inputs_for_repeat_dataset; in TEST_P()
[all …]
|
| D | take_dataset_op_test.cc | 50 OpKernel *op_kernel, gtl::InlinedVector<TensorValue, 4> *const inputs, in CreateTakeDatasetContext() argument
166 gtl::InlinedVector<TensorValue, 4> inputs_for_take_dataset; in TEST_P()
217 gtl::InlinedVector<TensorValue, 4> inputs_for_take_dataset; in TEST_F()
247 gtl::InlinedVector<TensorValue, 4> inputs_for_take_dataset; in TEST_P()
278 gtl::InlinedVector<TensorValue, 4> inputs_for_take_dataset; in TEST_P()
309 gtl::InlinedVector<TensorValue, 4> inputs_for_take_dataset; in TEST_P()
339 gtl::InlinedVector<TensorValue, 4> inputs_for_take_dataset; in TEST_F()
374 gtl::InlinedVector<TensorValue, 4> inputs_for_take_dataset; in TEST_P()
412 gtl::InlinedVector<TensorValue, 4> inputs_for_take_dataset; in TEST_P()
450 gtl::InlinedVector<TensorValue, 4> inputs_for_take_dataset; in TEST_P()
[all …]
|
| D | dataset_test_base.cc | 74 gtl::InlinedVector<TensorValue, 4> inputs; in CreateTensorSliceDataset()
179 OpKernel* kernel, gtl::InlinedVector<TensorValue, 4>* inputs, in CreateOpKernelContext() argument
221 const OpKernel& kernel, const gtl::InlinedVector<TensorValue, 4>& inputs) { in CheckOpKernelInput() argument
231 gtl::InlinedVector<TensorValue, 4>* inputs, DataTypeVector input_types, in AddDatasetInput() argument
|
| D | single_threaded_executor.cc | 28 typedef gtl::InlinedVector<TensorValue, 4> TensorValueVec;
295 TensorValue val = ctx.release_output(j); in RunAsync()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.h | 55 const gtl::InlinedVector<TensorValue, 4>& inputs,
57 gtl::InlinedVector<TensorValue, 4>* output);
|
| D | constant_folding.h | 40 static Status CreateNodeDef(const string& name, const TensorValue& tensor,
77 const gtl::InlinedVector<TensorValue, 4>& inputs,
78 gtl::InlinedVector<TensorValue, 4>* output) const;
166 const gtl::InlinedVector<TensorValue, 4>& reduction_indices_vector) const;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 513 struct TensorValue { struct
514 TensorValue() : mutex_if_ref(nullptr), tensor(nullptr) {} in TensorValue() argument
515 TensorValue(Tensor* t) // NOLINT(runtime/explicit) in TensorValue() argument
517 TensorValue(mutex* mu, Tensor* t) : mutex_if_ref(mu), tensor(t) {} in TensorValue() function
651 const gtl::InlinedVector<TensorValue, 4>* inputs = nullptr;
1183 TensorValue release_output(int index);
1263 gtl::InlinedVector<TensorValue, 4> outputs_;
1502 const TensorValue& value((*params_->inputs)[index]); in input_dtype()
1529 const TensorValue& value((*params_->inputs)[index]); in input_is_ref()
1577 inline TensorValue OpKernelContext::release_output(int index) { in release_output()
[all …]
|
| D | op_kernel.cc | 302 for (TensorValue& value : outputs_) { in ~OpKernelContext()
378 const TensorValue& value((*params_->inputs)[start]); in input_dtype()
460 outputs_[output_index] = TensorValue(new_tensor.release()); in forward_input_to_output_with_shape()
502 const TensorValue& input = (*params_->inputs)[input_index]; in forward_input()
736 outputs_[index] = TensorValue(output_tensor.release()); in allocate_output()
802 outputs_[index] = TensorValue(new Tensor(tensor)); in set_output()
827 outputs_[index] = TensorValue(mu, tensor_for_ref); in set_output_ref()
875 for (const TensorValue& t : *params_->inputs) { in MatchSignature()
|
| D | op_kernel_test.cc | 389 gtl::InlinedVector<TensorValue, 4> inputs{TensorValue(&a), TensorValue(&b), in TEST_F()
390 TensorValue(&c)}; in TEST_F()
669 gtl::InlinedVector<TensorValue, 4> inputs{}; in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.h | 87 absl::InlinedVector<TensorValue, 4> tensor_inputs_;
|
| D | graph_compiler.cc | 110 using NodeOutputs = std::vector<TensorValue>; in Compile()
114 for (const TensorValue& value : outputs) { in Compile()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_util.h | 40 gtl::InlinedVector<TensorValue, 4> sub_inputs_;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.h | 194 TensorValue mutable_input(int input_index) { in mutable_input()
237 gtl::InlinedVector<TensorValue, 4> inputs_;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | message_wrappers.h | 619 virtual Status TensorValue(size_t i, Tensor* out_tensor) const = 0;
657 Status TensorValue(size_t i, Tensor* out_tensor) const override;
686 Status TensorValue(size_t i, Tensor* out_tensor) const override;
711 Status TensorValue(size_t i, Tensor* out_tensor) const override;
|
| D | message_wrappers_test.cc | 154 TF_EXPECT_OK(response.TensorValue(0, &val)); in CheckRunStepResponse()
156 TF_EXPECT_OK(response.TensorValue(1, &val)); in CheckRunStepResponse()
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 64 gtl::InlinedVector<TensorValue, 4> inputs; in TestBitcastOp()
|