Home
last modified time | relevance | path

Searched defs:input_shape_or (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbincount_op.cc62 StatusOr<xla::Shape> input_shape_or = ctx->builder()->GetShape(input); in Compile() local
H A Dunique_op.cc147 StatusOr<xla::Shape> input_shape_or = ctx->builder()->GetShape(input); in CompileWithAxis() local
H A Dwhile_op.cc517 auto input_shape_or = ctx->builder()->GetShape(input); in Compile() local