/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() function 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()
|