Searched refs:SetTensorValue (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 452 TF_ASSERT_OK(SetTensorValue(t.dtype(), 2, &t)); in TEST_F()
458 TF_ASSERT_OK(SetTensorValue(t.dtype(), std::numeric_limits<int>::max(), &t)); in TEST_F()
465 TF_ASSERT_OK(SetTensorValue(t.dtype(), std::numeric_limits<int>::min(), &t)); in TEST_F()
481 Status s = SetTensorValue(t.dtype(), val, &t); in TestSetTensorValue()
|
| D | utils.h | 299 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 | 2302 if (!SetTensorValue(DT_INT32, axis, &axis_t).ok() || in SimplifyPack()
2399 if (!already_optimized && SetTensorValue(DT_BOOL, true, &true_t).ok() && in SimplifySwitch()
2400 SetTensorValue(DT_BOOL, false, &false_t).ok()) { in SimplifySwitch()
|
| D | arithmetic_optimizer.cc | 2423 Status status = SetTensorValue(type, num_inputs, &t); in TrySimplify()
|