Searched refs:SetTensorValue (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/ |
D | utils_test.cc | 525 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()
|
D | utils.h | 398 Status SetTensorValue(DataType dtype, int value, Tensor* tensor);
|
D | utils.cc | 442 Status SetTensorValue(DataType dtype, int value, Tensor* tensor) { in SetTensorValue() function
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | constant_folding.cc | 2440 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()
|
D | arithmetic_optimizer.cc | 2530 Status status = SetTensorValue(type, num_inputs, &t); in TrySimplify()
|