Searched defs:counter_shape (Results 1 – 3 of 3) sorted by relevance
93 auto counter_shape = in GetKeyCounter() local191 TensorShape const& counter_shape, in MaybeSliceCounter()231 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile() local300 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile() local364 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile() local414 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile() local484 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile() local
90 const xla::Shape counter_shape = xla::ShapeUtil::MakeScalarShape(xla::S32); in RollingSelectR1() local
26 TensorShape const& counter_shape) { in CheckKeyCounterShape()