Searched refs:depth_shape (Results 1 – 1 of 1) sorted by relevance
34 const TensorShape depth_shape = ctx->InputShape(1); in Compile() local46 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(depth_shape), in Compile()48 depth_shape.DebugString())); in Compile()