Searched defs:dy_shape (Results 1 – 3 of 3) sorted by relevance
54 def _compute_theoretical_jacobian(x, x_shape, x_data, dy, dy_shape, dx, argument
240 TensorShape dy_shape = context->input(4).shape(); in Compute() local
551 const TensorShape dy_shape = ctx->InputShape(4); in Compile() local