Searched refs:MlirTensor (Results 1 – 1 of 1) sorted by relevance
93 class MlirTensor : public TracingTensorHandle { class95 explicit MlirTensor(Value value) in MlirTensor() function in mlir::TF::__anon2b880b700111::MlirTensor551 retvals[i] = new MlirTensor(op->getResult(i)); in Execute()567 *handle = new MlirTensor(func_.getBody().front().addArgument(type)); in AddParameter()578 auto* operand = dyn_cast<MlirTensor>(input); in AddInput()597 expected_type = cast<MlirTensor>(input)->getElementType(); in AddInput()613 auto* operand = dyn_cast<MlirTensor>(input); in AddInputList()634 if (arg_def_type != cast<MlirTensor>(input)->getElementType()) in AddInputList()643 TypeAttr::get(cast<MlirTensor>(inputs.front())->getElementType()); in AddInputList()650 types.push_back(TypeAttr::get(cast<MlirTensor>(input)->getElementType())); in AddInputList()[all …]