Home
last modified time | relevance | path

Searched refs:TensorValue (Results 1 – 25 of 50) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h81 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,
Dkernel_and_device_test.cc131 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()
Dkernel_and_device.cc151 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()
Dtensor_handle.h82 Status TensorValue(tensorflow::TensorValue* t);
Dtensor_handle.cc151 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/
Dtensor_dataset_op_test.cc57 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 …]
Ddataset_test_base.h94 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,
Dsparse_tensor_slice_dataset_op_test.cc51 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 …]
Dtensor_slice_dataset_op_test.cc59 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 …]
Drepeat_dataset_op_test.cc53 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 …]
Dtake_dataset_op_test.cc50 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 …]
Ddataset_test_base.cc74 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
Dsingle_threaded_executor.cc28 typedef gtl::InlinedVector<TensorValue, 4> TensorValueVec;
295 TensorValue val = ctx.release_output(j); in RunAsync()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.h55 const gtl::InlinedVector<TensorValue, 4>& inputs,
57 gtl::InlinedVector<TensorValue, 4>* output);
Dconstant_folding.h40 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/
Dop_kernel.h513 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 …]
Dop_kernel.cc302 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()
Dop_kernel_test.cc389 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/
Dgraph_compiler.h87 absl::InlinedVector<TensorValue, 4> tensor_inputs_;
Dgraph_compiler.cc110 using NodeOutputs = std::vector<TensorValue>; in Compile()
114 for (const TensorValue& value : outputs) { in Compile()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_util.h40 gtl::InlinedVector<TensorValue, 4> sub_inputs_;
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.h194 TensorValue mutable_input(int input_index) { in mutable_input()
237 gtl::InlinedVector<TensorValue, 4> inputs_;
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.h619 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;
Dmessage_wrappers_test.cc154 TF_EXPECT_OK(response.TensorValue(0, &val)); in CheckRunStepResponse()
156 TF_EXPECT_OK(response.TensorValue(1, &val)); in CheckRunStepResponse()
/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op_test.cc64 gtl::InlinedVector<TensorValue, 4> inputs; in TestBitcastOp()

12