Searched refs:imm_ctx (Results 1 – 2 of 2) sorted by relevance
74 auto imm_ctx = dyn_cast<ImmediateExecutionContext>(unwrap(ctx)); in TF_ExecutionContextGetTFEContext() local75 if (!imm_ctx) { in TF_ExecutionContextGetTFEContext()81 return wrap(imm_ctx); in TF_ExecutionContextGetTFEContext()
80 auto imm_ctx = dyn_cast<ImmediateExecutionContext>(ctx); in BroadcastMul() local81 AbstractTensorPtr minus_1(imm_ctx->CreateInt32Scalar(-1)); in BroadcastMul()82 ImmediateTensorHandlePtr dim(imm_ctx->CreateLocalHandle(minus_1.get())); in BroadcastMul()