Home
last modified time | relevance | path

Searched refs:num_upper_in_shape (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_band_part_op.cc44 const TensorShape num_upper_in_shape = context->InputShape(2); in Compile() local
45 OP_REQUIRES(context, TensorShapeUtils::IsScalar(num_upper_in_shape), in Compile()
47 num_upper_in_shape.DebugString())); in Compile()