Searched defs:grad_shape (Results 1 – 4 of 4) sorted by relevance
141 TensorShape grad_shape = ctx->InputShape(3); in Compile() local199 TensorShape grad_shape = ctx->InputShape(3); in Compile() local257 TensorShape grad_shape = ctx->InputShape(3); in Compile() local311 TensorShape grad_shape = ctx->InputShape(4); in Compile() local367 TensorShape grad_shape = ctx->InputShape(5); in Compile() local417 TensorShape grad_shape = ctx->InputShape(3); in Compile() local488 TensorShape grad_shape = ctx->InputShape(9); in Compile() local583 TensorShape grad_shape = ctx->InputShape(8); in Compile() local670 TensorShape grad_shape = ctx->InputShape("grad"); in Compile() local770 TensorShape grad_shape = ctx->InputShape(3); in CompileFtrl() local[all …]
414 xla::Shape grad_shape = ctx->InputXlaShape(4).ValueOrDie(); in CompileAsDynamicUpdateSlice() local566 xla::Shape grad_shape = ctx->builder()->GetShape(grad).ValueOrDie(); in Compile() local
655 TensorShape grad_shape = ctx->InputShape(0); in Compile() local
1474 grad_shape=None, argument