Searched refs:stride_shape (Results 1 – 1 of 1) sorted by relevance
105 const TensorShape stride_shape = ctx->InputShape(2); in GetStride() local107 if (!TensorShapeUtils::IsVector(stride_shape)) { in GetStride()109 stride_shape.DebugString()); in GetStride()111 if (stride_shape.num_elements() != num_dims()) { in GetStride()298 const TensorShape stride_shape = ctx->InputShape(4); in Compile() local300 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(stride_shape), in Compile()302 stride_shape.DebugString())); in Compile()518 const TensorShape stride_shape = ctx->InputShape(4); in Compile() local520 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(stride_shape), in Compile()522 stride_shape.DebugString())); in Compile()