Searched refs:dynamic_sizes (Results 1 – 2 of 2) sorted by relevance
246 auto dynamic_sizes = operands.begin(); in fillShapeFromAllocLike() local249 ? ValueOrConst{*(dynamic_sizes++)} in fillShapeFromAllocLike()
54 std::vector<int64> dynamic_sizes; in GetTensorListDynamicDims() local56 TF_RETURN_IF_ERROR(ctx->ConstantInputAsIntVector(0, &dynamic_sizes)); in GetTensorListDynamicDims()80 xla::ConstantR0<int32>(ctx->builder(), dynamic_sizes[dim])); in GetTensorListDynamicDims()