Searched refs:start_tensor (Results 1 – 3 of 3) sorted by relevance
258 Tensor<int, 1> start_tensor(10); in test_compound_assign() local260 start_tensor.setRandom(); in test_compound_assign()263 Tensor<int, 1> tensor = start_tensor; in test_compound_assign()266 VERIFY_IS_EQUAL(tensor(i), start_tensor(i) + offset_tensor(i)); in test_compound_assign()269 tensor = start_tensor; in test_compound_assign()272 VERIFY_IS_EQUAL(tensor(i), start_tensor(i) - offset_tensor(i)); in test_compound_assign()275 tensor = start_tensor; in test_compound_assign()278 VERIFY_IS_EQUAL(tensor(i), start_tensor(i) * offset_tensor(i)); in test_compound_assign()281 tensor = start_tensor; in test_compound_assign()284 VERIFY_IS_EQUAL(tensor(i), start_tensor(i) / offset_tensor(i)); in test_compound_assign()
2464 auto start_tensor = operands[0].dyn_cast_or_null<ElementsAttr>(); in fold() local2467 if (start_tensor && limit_tensor && delta_tensor) { in fold()2469 assert(start_tensor.getType().getRank() == 0 && in fold()2474 auto start_attr = start_tensor.getValue<IntegerAttr>({}); in fold()2482 auto start_attr = start_tensor.getValue<FloatAttr>({}); in fold()
987 Tensor start_tensor = CreateTensor<int64>(TensorShape({}), {start_}); in GetInputTensors() local990 return {start_tensor, stop_tensor, step_tensor}; in GetInputTensors()