Searched defs:output_dim_size (Results 1 – 7 of 7) sorted by relevance
107 const int64 output_dim_size = output_shape[j]; in SparseConcatPermuteKernel() local169 int64 output_dim_size = in operator ()() local
34 const int output_dim_size = context->tensors[outputs->data[0]].dims->size; in PopulateSubGraph() local
107 llvm::Value* output_dim_size = llvm::ConstantInt::get( in EmitDynamicUpdateSliceInPlaceImpl() local
1161 const int64_t output_dim_size = in HandleReshape() local
2169 int64_t output_dim_size = in EmitElementalGather() local
2522 int64_t output_dim_size = in HandleGather() local
2701 int64_t output_dim_size = hlo->shape().dimensions(output_sharded_dim); in HandleReshape() local