Home
last modified time | relevance | path

Searched refs:SetTensorValue (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc525 TF_ASSERT_OK(SetTensorValue(t.dtype(), 2, &t)); in TEST_F()
531 TF_ASSERT_OK(SetTensorValue(t.dtype(), std::numeric_limits<int>::max(), &t)); in TEST_F()
538 TF_ASSERT_OK(SetTensorValue(t.dtype(), std::numeric_limits<int>::min(), &t)); in TEST_F()
600 Status s = SetTensorValue(t.dtype(), val, &t); in TestSetTensorValue()
Dutils.h398 Status SetTensorValue(DataType dtype, int value, Tensor* tensor);
Dutils.cc442 Status SetTensorValue(DataType dtype, int value, Tensor* tensor) { in SetTensorValue() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc2440 if (!SetTensorValue(DT_INT32, axis, &axis_t).ok() || in SimplifyPack()
2659 if (!already_optimized && SetTensorValue(DT_BOOL, true, &true_t).ok() && in SimplifySwitch()
2660 SetTensorValue(DT_BOOL, false, &false_t).ok()) { in SimplifySwitch()
Darithmetic_optimizer.cc2530 Status status = SetTensorValue(type, num_inputs, &t); in TrySimplify()