Searched defs:counter_shape (Results 1 – 2 of 2) sorted by relevance
91 auto counter_shape = in GetKeyCounter() local185 TensorShape const& counter_shape, in MaybeSliceCounter()217 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile() local270 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile() local331 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile() local377 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile() local431 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile() local
26 TensorShape const& counter_shape) { in CheckKeyCounterShape()