Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node.cc40 packed_args_.emplace(index, gtl::InlinedVector<TensorValue, 4>(num_handles)); in InitPackedHandle()
41 TensorValue* packed_arg_flat = &(packed_args_[index][0]); in InitPackedHandle()
47 const Status status = h->TensorValue(h->device(), &packed_arg_flat[i]); in InitPackedHandle()
75 TensorValue* tensor_args_flat = &tensor_args_[0]; in Init()
79 Status s = in->TensorValue(ctx->CanonicalDevice(d), &tensor_args_flat[i]); in Init()
Dtensor_handle_data.cc33 Status LocalTensorHandleData::TensorValue(tensorflow::TensorValue* t) { in TensorValue() function in tensorflow::LocalTensorHandleData
37 *t = tensorflow::TensorValue(&tensor); in TensorValue()
Dkernel_and_device_test.cc126 gtl::InlinedVector<TensorValue, 4> inputs; in BM_KernelAndDeviceRun()
127 inputs.push_back(TensorValue(&t)); in BM_KernelAndDeviceRun()
128 inputs.push_back(TensorValue(&t)); in BM_KernelAndDeviceRun()
Dtensor_handle_data.h36 Status TensorValue(tensorflow::TensorValue* t);
Dkernel_and_device.h68 explicit EagerKernelArgs(gtl::InlinedVector<TensorValue, 4>&& tensor_args) in EagerKernelArgs() argument
74 TensorValue* MutableInput(int i) { return &tensor_args_[i]; } in MutableInput()
80 const gtl::InlinedVector<TensorValue, 4>* GetTensorValues() const { in GetTensorValues()
85 gtl::InlinedVector<TensorValue, 4> tensor_args_;
Dtensor_handle.h148 Status TensorValue(const Device* d, tensorflow::TensorValue* t);
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.h54 const gtl::InlinedVector<TensorValue, 4>& inputs,
56 gtl::InlinedVector<TensorValue, 4>* output);
Dconstant_folding.h43 static Status CreateNodeDef(const string& name, const TensorValue& tensor,
95 const gtl::InlinedVector<TensorValue, 4>& inputs,
96 gtl::InlinedVector<TensorValue, 4>* output) const;
238 const gtl::InlinedVector<TensorValue, 4>& reduction_indices_vector) const;
Devaluation_utils.cc27 using TensorVector = gtl::InlinedVector<TensorValue, 4>;
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op_test.cc108 gtl::InlinedVector<TensorValue, 4>* const inputs, in CreateMapDefunContext() argument
245 gtl::InlinedVector<TensorValue, 4> input_values; in TEST_P()
247 input_values.push_back(TensorValue(&input)); in TEST_P()
274 gtl::InlinedVector<TensorValue, 4> input_values; in TEST_F()
276 input_values.push_back(TensorValue(&input)); in TEST_F()
Ddataset_test_base.cc320 gtl::InlinedVector<TensorValue, 4>* const inputs, in CreateDatasetContext() argument
484 OpKernel* kernel, gtl::InlinedVector<TensorValue, 4>* inputs, in CreateOpKernelContext() argument
490 OpKernel* kernel, gtl::InlinedVector<TensorValue, 4>* inputs, in CreateOpKernelContext() argument
534 const OpKernel& kernel, const gtl::InlinedVector<TensorValue, 4>& inputs) { in CheckOpKernelInput() argument
544 gtl::InlinedVector<TensorValue, 4>* inputs, DataTypeVector input_types, in AddDatasetInput() argument
821 gtl::InlinedVector<TensorValue, 4> inputs; in RunDatasetOp()
823 inputs.emplace_back(TensorValue(input_dataset)); in RunDatasetOp()
830 inputs.push_back(TensorValue(copy.get())); in RunDatasetOp()
926 gtl::InlinedVector<TensorValue, 4> inputs; in MakeDatasetTensor()
928 inputs.emplace_back(TensorValue(input_dataset)); in MakeDatasetTensor()
[all …]
Ddataset_test_base.h662 gtl::InlinedVector<TensorValue, 4>* const inputs,
694 const gtl::InlinedVector<TensorValue, 4>& inputs);
698 gtl::InlinedVector<TensorValue, 4>* inputs,
703 gtl::InlinedVector<TensorValue, 4>* inputs,
748 Status AddDatasetInput(gtl::InlinedVector<TensorValue, 4>* inputs,
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_util.cc85 sub_inputs_({TensorValue(output), TensorValue(input)}), in SubContext()
Dcollective_util.h40 gtl::InlinedVector<TensorValue, 4> sub_inputs_;
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h510 struct TensorValue { struct
511 TensorValue() : mutex_if_ref(nullptr), tensor(nullptr) {} in TensorValue() function
512 explicit TensorValue(Tensor* t) : mutex_if_ref(nullptr), tensor(t) {} in TensorValue() function
513 TensorValue(mutex* mu, Tensor* t) : mutex_if_ref(mu), tensor(t) {} in TensorValue() function
671 const gtl::InlinedVector<TensorValue, 4>* inputs = nullptr;
1214 TensorValue release_output(int index);
1304 gtl::InlinedVector<TensorValue, 4> outputs_;
1581 const TensorValue& value((*params_->inputs)[index]); in input_dtype()
1604 const TensorValue& value((*params_->inputs)[index]); in input_is_ref()
1628 inline TensorValue OpKernelContext::release_output(int index) { in release_output()
[all …]
Dop_kernel_test.cc372 gtl::InlinedVector<TensorValue, 4> inputs{TensorValue(&a), TensorValue(&b), in TEST_F()
373 TensorValue(&c)}; in TEST_F()
742 gtl::InlinedVector<TensorValue, 4> inputs{}; in TEST_F()
1096 gtl::InlinedVector<TensorValue, 4> inputs{TensorValue(&a), TensorValue(&b)}; in BM_TraceString()
Dop_kernel.cc348 for (TensorValue& value : outputs_) { in ~OpKernelContext()
407 const TensorValue& value((*params_->inputs)[index]); in input_dtype()
476 outputs_[output_index] = TensorValue(new_tensor.release()); in forward_input_to_output_with_shape()
504 const TensorValue& input = (*params_->inputs)[input_index]; in forward_input()
769 outputs_[index] = TensorValue(output_tensor.release()); in allocate_output()
942 outputs_[index] = TensorValue(new_tensor.release()); in maybe_set_output_by_allocate_and_copy()
975 outputs_[index] = TensorValue(new Tensor(tensor)); in set_output()
988 outputs_[index] = TensorValue(new Tensor(std::move(tensor))); in set_output()
998 outputs_[index] = TensorValue(mu, tensor_for_ref); in set_output_ref()
1034 for (const TensorValue& t : *params_->inputs) { in MatchSignature()
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.h142 TensorValue mutable_input(int input_index);
174 gtl::InlinedVector<TensorValue, 4> inputs_;
Dfingerprint_op_test.cc52 inputs_.push_back(TensorValue(data)); in MakeFingerprintOp()
56 inputs_.push_back(TensorValue(&method_)); in MakeFingerprintOp()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.h87 absl::InlinedVector<TensorValue, 4> tensor_inputs_;
Dgraph_compiler.cc116 using NodeOutputs = std::vector<TensorValue>; in Compile()
120 for (const TensorValue& value : outputs) { in Compile()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc714 gtl::InlinedVector<TensorValue, 4>&& tensor_args, in TestExecuteNodeArgs() argument
873 gtl::InlinedVector<TensorValue, 4> tensor_args = {TensorValue()}; in TEST_F()
965 gtl::InlinedVector<TensorValue, 4> input_tensors = {TensorValue()}; in TEST_F()
1013 gtl::InlinedVector<TensorValue, 4> input_tensors = {TensorValue()}; in TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.h631 virtual Status TensorValue(size_t i, Tensor* out_tensor) const = 0;
669 Status TensorValue(size_t i, Tensor* out_tensor) const override;
698 Status TensorValue(size_t i, Tensor* out_tensor) const override;
723 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/core/common_runtime/device/
Ddevice_event_mgr_test.cc185 gtl::InlinedVector<TensorValue, 4> add_inputs_;
312 add_inputs_[0] = TensorValue(&gpu_inputs_[0]); in PrepOpKernel()
313 add_inputs_[1] = TensorValue(&gpu_inputs_[1]); in PrepOpKernel()

123