training_ops.cc | 36 TensorShape var_shape; variable 37 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &handle)); 46 ctx, var_shape.IsSameSize(delta_shape), 48 var_shape.DebugString(), " vs ", 70 TensorShape var_shape, accum_shape; variable 72 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &var)); 75 OP_REQUIRES(ctx, var_shape.IsSameSize(accum_shape), 78 var_shape.DebugString(), " ", accum_shape.DebugString())); 86 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape), 89 var_shape.DebugString(), " ", grad_shape.DebugString())) 128 TensorShape var_shape, accum_shape; variable 170 TensorShape var_shape, m_shape, v_shape; variable 262 TensorShape var_shape, ms_shape, mom_shape; variable 346 TensorShape var_shape, accum_shape, linear_shape; local [all...] |