| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| bias_ops.cc | 91 const TensorShape out_backprop_shape = ctx->InputShape(0); variable
93 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrixOrHigher(out_backprop_shape),
95 out_backprop_shape.DebugString()));
99 ? out_backprop_shape.dims() - 1
106 std::vector<int64> reduce_dims(out_backprop_shape.dims() - 1);
|
| pooling_ops.cc | 310 const TensorShape out_backprop_shape = ctx->InputShape(2); variable
320 OP_REQUIRES(ctx, out_backprop_shape.dims() == num_dims(),
407 const TensorShape out_backprop_shape = ctx->InputShape(1); variable
415 OP_REQUIRES(ctx, out_backprop_shape.dims() == num_dims(),
517 const TensorShape out_backprop_shape = ctx->InputShape(2); variable
527 OP_REQUIRES(ctx, out_backprop_shape.dims() == num_dims(),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_avgpooling_op.cc | 247 GetMklShape(context, 1, &mkl_context.out_backprop_shape);
249 mkl_context.out_backprop_shape.IsMklTensor();
264 mkl_context.params.in_dim = mkl_context.out_backprop_shape.GetDimension();
294 mkl_context.out_backprop_shape.GetConvertedFlatData(
346 MklShape out_backprop_shape; member in struct:tensorflow::MklAvgPoolingGradOp::__anon45423
355 bool outbackprop_in_mkl_format = out_backprop_shape.IsMklTensor();
374 context, out_backprop_shape.GetDimension() == 4,
385 lt_user_outbackprop = (dnnLayout_t)out_backprop_shape.GetCurLayout();
414 bool outbackprop_in_mkl_format = out_backprop_shape.IsMklTensor();
|
| conv_grad_ops_3d.cc | 221 const TensorShape& out_backprop_shape = out_backprop.shape(); variable
235 input_shape, filter_shape, out_backprop_shape,
327 const TensorShape& out_backprop_shape = out_backprop.shape(); variable
341 input_shape, filter_shape, out_backprop_shape,
416 out_backprop_shape.num_elements();
677 const TensorShape& out_backprop_shape = out_backprop.shape(); variable
692 input_shape, filter_shape, out_backprop_shape, stride_,
789 const TensorShape& out_backprop_shape = out_backprop.shape(); variable
1143 const TensorShape& out_backprop_shape = out_backprop.shape(); variable
1527 const TensorShape& out_backprop_shape = out_backprop.shape(); variable
[all...] |
