Searched defs:out_backprop_shape (Results 1 – 6 of 6) sorted by relevance
100 const TensorShape& filter_shape, const TensorShape& out_backprop_shape, in ConvBackpropComputeDimensionsV2()162 const TensorShape& out_backprop_shape, in ConvBackpropComputeDimensions()
221 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local327 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local677 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local789 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local1143 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local1527 const TensorShape& out_backprop_shape = out_backprop.shape(); in Compute() local
346 MklShape out_backprop_shape; member
91 const TensorShape out_backprop_shape = ctx->InputShape(0); in Compile() local
310 const TensorShape out_backprop_shape = ctx->InputShape(2); in Compile() local407 const TensorShape out_backprop_shape = ctx->InputShape(1); in Compile() local517 const TensorShape out_backprop_shape = ctx->InputShape(2); in Compile() local
204 const xla::Shape& filter_shape, const xla::Shape& out_backprop_shape, in ConvBackpropComputeDimensionsV2XlaShapes()