Searched defs:out_backprop_shape (Results 1 – 6 of 6) sorted by relevance
96 const TensorShape& filter_shape, const TensorShape& out_backprop_shape, in ConvBackpropComputeDimensionsV2()158 const TensorShape& out_backprop_shape, in ConvBackpropComputeDimensions()171 const TensorShape& out_backprop_shape, in Conv2DBackpropComputeInputShape()
230 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local337 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local688 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local800 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local1156 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local1638 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local
91 const TensorShape out_backprop_shape = ctx->InputShape(0); in Compile() local
317 const TensorShape out_backprop_shape = ctx->InputShape(2); in Compile() local428 const TensorShape out_backprop_shape = ctx->InputShape(1); in Compile() local537 const TensorShape out_backprop_shape = ctx->InputShape(2); in Compile() local
138 const xla::Shape& filter_shape, const xla::Shape& out_backprop_shape, in ConvBackpropComputeDimensionsV2XlaShapes()
4303 ArrayRef<int64_t> out_backprop_shape = out_backprop_ty.getShape(); in matchAndRewrite() local