Searched refs:generic_input (Results 1 – 4 of 4) sorted by relevance
115 for (const ImmediateExecutionTensorHandle* generic_input : op.GetInputs()) { in MaybePinToCustomDevice() local118 if (CustomDeviceTensorHandle::classof(generic_input)) { in MaybePinToCustomDevice()120 down_cast<const CustomDeviceTensorHandle*>(generic_input); in MaybePinToCustomDevice()138 for (const ImmediateExecutionTensorHandle* generic_input : op.GetInputs()) { in MaybePinToCustomDevice() local139 if (generic_input->DataType() == DT_RESOURCE) { in MaybePinToCustomDevice()140 if (CustomDeviceTensorHandle::classof(generic_input)) { in MaybePinToCustomDevice()142 down_cast<const CustomDeviceTensorHandle*>(generic_input); in MaybePinToCustomDevice()
116 %generic_input = addrspacecast float addrspace(1)* %input to float*117 %addr = getelementptr float, float* %generic_input, i64 10
124 %generic_input = addrspacecast float addrspace(1)* %input to float*125 %addr = getelementptr float, float* %generic_input, i64 10
98 %generic_input = addrspacecast float addrspace(1)* %input to float*99 %addr = getelementptr float, float* %generic_input, i64 10