Searched defs:out_backprop_shape (Results 1 – 5 of 5) sorted by relevance
97 const TensorShape& filter_shape, const TensorShape& out_backprop_shape, in ConvBackpropComputeDimensionsV2()163 const TensorShape& out_backprop_shape, in ConvBackpropComputeDimensions()176 const TensorShape& out_backprop_shape, in Conv2DBackpropComputeInputShape()
234 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local363 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local741 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local879 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local1268 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local1648 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local
365 const TensorShape out_backprop_shape = ctx->InputShape(2); in Compile() local476 const TensorShape out_backprop_shape = ctx->InputShape(1); in Compile() local589 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()
5110 ArrayRef<int64_t> out_backprop_shape = out_backprop_ty.getShape(); in matchAndRewrite() local