Searched refs:delta_input_idx (Results 1 – 2 of 2) sorted by relevance
269 auto delta_input_idx = 2; in Compute() local274 ctx, GetScalar(ctx->input(delta_input_idx), delta_input_idx, &delta_)); in Compute()
435 const int delta_input_idx = 2; in Compile() local462 xla::XlaOp delta = ctx->Input(delta_input_idx); in Compile()